From ae241a9af586f0ebd6bb418aa0e2416b81bf4215 Mon Sep 17 00:00:00 2001
From: turboderp <11859846+turboderp@users.noreply.github.com>
Date: Mon, 30 Dec 2024 02:24:49 +0100
Subject: [PATCH] Fix video example

---
 examples/multimodal_video.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/multimodal_video.py b/examples/multimodal_video.py
index 3fa5f853..21a552b6 100644
--- 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."