Fix sam link
Browse files
app.py
CHANGED
|
@@ -253,7 +253,7 @@ def build_model_lora(pipe_concept, lora_paths, style_path, condition, args, pipe
|
|
| 253 |
def build_yolo_segment_model(sam_path, device):
|
| 254 |
yolo_world = YOLOWorld(model_id="yolo_world/l")
|
| 255 |
sam = EfficientViTSamPredictor(
|
| 256 |
-
create_sam_model(name=
|
| 257 |
)
|
| 258 |
return yolo_world, sam
|
| 259 |
|
|
|
|
| 253 |
def build_yolo_segment_model(sam_path, device):
|
| 254 |
yolo_world = YOLOWorld(model_id="yolo_world/l")
|
| 255 |
sam = EfficientViTSamPredictor(
|
| 256 |
+
create_sam_model(name='efficientvit_sam_xl1.pt', weight_url=sam_path).to(device).eval()
|
| 257 |
)
|
| 258 |
return yolo_world, sam
|
| 259 |
|