diff --git a/examples/vlm/llava_next_generation_example.py b/examples/vlm/llava_next_generation_example.py index 4c15a687d156..de81c1cc181b 100644 --- a/examples/vlm/llava_next_generation_example.py +++ b/examples/vlm/llava_next_generation_example.py @@ -47,7 +47,7 @@ def main(args) -> None: # Tokenize the input texts # processor = AutoProcessor.from_pretrained("llava-hf/llava-1.5-7b-hf") processor = AutoProcessor.from_pretrained("llava-hf/llava-v1.6-vicuna-7b-hf") - data_path = '/lustre/fsw/coreai_dlalgo_genai/datasets/energon_datasets/LLaVA-Pretrain-LCS-558K' + data_path = '/home/ykarnati/Downloads/LLaVA-Pretrain/wds' image_processor = processor.image_processor tokenizer = processor.tokenizer