small change in llava next example

NVIDIA · Oct 24, 2024 · 4ded144 · 4ded144
1 parent ea8ecd1
commit 4ded144
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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