Update README.md

README.md

@@ -74,12 +74,12 @@ Instantiate the tokenizer, processor, and model.
 ```python
 device = torch.device("cuda:0")
 
-tokenizer = AutoTokenizer.from_pretrained("togethercomputer/Llama-3.1-8B-Dragonfly-v2")
+tokenizer = AutoTokenizer.from_pretrained("togethercomputer/Llama-3.1-8B-Dragonfly-Med-v2")
 clip_processor = AutoProcessor.from_pretrained("openai/clip-vit-large-patch14-336")
 image_processor = clip_processor.image_processor
 processor = DragonflyProcessor(image_processor=image_processor, tokenizer=tokenizer, image_encoding_style="llava-hd")
 
-model = DragonflyForCausalLM.from_pretrained("togethercomputer/Llama-3.1-8B-Dragonfly-v2")
+model = DragonflyForCausalLM.from_pretrained("togethercomputer/Llama-3.1-8B-Dragonfly-Med-v2")
 model = model.to(torch.bfloat16)
 model = model.to(device)
 ```