Spaces:

LPX55
/

Lightning-Painter-Multitool

Running on Zero

LPX55 commited on 9 days ago

Commit

6fbb74e

1 Parent(s): a232a77

hold

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,8 @@ from PIL import Image, ImageDraw
 import numpy as np
 from sam2.sam2_image_predictor import SAM2ImagePredictor
-predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2.1-hiera-large")
-print(predictor)
 # with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
 #     predictor.set_image(<your_image>)
 #     masks, _, _ = predictor.predict(<input_prompts>)

 import numpy as np
 from sam2.sam2_image_predictor import SAM2ImagePredictor
+# predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2.1-hiera-large", device="cuda")
+# print(predictor)
 # with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
 #     predictor.set_image(<your_image>)
 #     masks, _, _ = predictor.predict(<input_prompts>)