Update README.md

README.md

@@ -26,7 +26,7 @@ import torch
 from transformers import AutoModelForCausalLM, AutoProcessor
 from open_flamingo.eval.models.cruise_model import EvalModel
 
-processor = AutoProcessor.from_pretrained("infimm/infimm-hd", trust_remote_code=True)
+processor = AutoProcessor.from_pretrained("Infi-MM/infimm-hd", trust_remote_code=True)
 
 prompts = [
     {
@@ -40,7 +40,7 @@ prompts = [
 inputs = processor(prompts)
 # use bf16 and gpu 0
 model = AutoModelForCausalLM.from_pretrained(
-    "infimm/infimm-hd",
+    "Infi-MM/infimm-hd",
     local_files_only=True,
     torch_dtype=torch.bfloat16,
     trust_remote_code=True,