Spaces:

LPX55
/

Lightning-Painter-Multitool

Running on Zero

LPX55 commited on Mar 29

Commit

1d11eb2

1 Parent(s): 6bc4886

wew

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ from sam2.sam2_image_predictor import SAM2ImagePredictor
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 SAM_MODEL = "facebook/sam2.1-hiera-large"
 MODELS = {
     "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",
     "Lustify Lightning": "GraydientPlatformAPI/lustify-lightning",

 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 SAM_MODEL = "facebook/sam2.1-hiera-large"
+PREDICTOR = None
 MODELS = {
     "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",
     "Lustify Lightning": "GraydientPlatformAPI/lustify-lightning",