ftopal commited on
Commit
21fceb9
·
verified ·
1 Parent(s): c9863b9

Fix sam link

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="xl1", weight_url=sam_path).to(device).eval()
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