Fix "Unresolved reference 'THIS_DIR' "

Add the value of THIS_DIR.
meta-llama · Nov 24, 2024 · 7e3f418 · 7e3f418
1 parent 362d52a
commit 7e3f418
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/models/scripts/multimodal_example_text_completion.py b/models/scripts/multimodal_example_text_completion.py
@@ -18,7 +18,9 @@
 
 from PIL import Image as PIL_Image
 from termcolor import cprint
+from pathlib import Path
 
+THIS_DIR = Path(__file__).parent.resolve()
 
 def run_main(
     ckpt_dir: str,