Fix video example

turboderp-org · Dec 30, 2024 · ae241a9 · ae241a9
1 parent 1ef6183
commit ae241a9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/multimodal_video.py b/examples/multimodal_video.py
@@ -31,10 +31,11 @@
 greedy = True
 
 model_directory = "/mnt/str/models/qwen2-vl-7b-instruct-exl2/6.0bpw"
+images_mask = os.path.join(os.path.dirname(os.path.abspath(__file__)), "media/test_video_*.png")
 
 frames = [
     {"file": f}
-    for f in sorted(glob.glob("media/test_video_*.png"))
+    for f in sorted(glob.glob(images_mask))
 ]
 
 instruction = "Describe this video."