diff --git a/models/scripts/multimodal_example_chat_completion.py b/models/scripts/multimodal_example_chat_completion.py index 46bbf828..de0a6106 100644 --- 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, diff --git a/models/scripts/multimodal_example_text_completion.py b/models/scripts/multimodal_example_text_completion.py index 8d18d7b0..40cb5134 100644 --- 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,