Fix "Unresolved reference 'THIS_DIR' "

Add the value of THIS_DIR.
meta-llama · Nov 24, 2024 · 6c55e59 · 6c55e59
1 parent 7e3f418
commit 6c55e59
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/models/scripts/multimodal_example_chat_completion.py b/models/scripts/multimodal_example_chat_completion.py
@@ -17,7 +17,9 @@
 from llama_models.llama3.reference_impl.generation import Llama
 
 from PIL import Image as PIL_Image
+from pathlib import Path
 
+THIS_DIR = Path(__file__).parent.resolve()
 
 def run_main(
     ckpt_dir: str,