SmolVLM2-XSPFGenerator

Sleeping

mfarre HF staff commited on Feb 13

Commit

3dd7eae

1 Parent(s): fbe9168

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,7 +53,7 @@ class VideoHighlightDetector:
         self.model = AutoModelForVision2Seq.from_pretrained(
             model_path,
             torch_dtype=torch.bfloat16,
-            attn_implementation="flash_attention_2"
         ).to(device)
     def analyze_video_content(self, video_path: str) -> str:

         self.model = AutoModelForVision2Seq.from_pretrained(
             model_path,
             torch_dtype=torch.bfloat16,
+            _attn_implementation="flash_attention_2"
         ).to(device)
     def analyze_video_content(self, video_path: str) -> str: