Update README.md

README.md

@@ -36,7 +36,7 @@ Cephalo is a series of multimodal materials science and engineering focused visi
 
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/623ce1c6b66fedf374859fe7/kl5GWBP9WS0D4uwd1t3S7.png)
 
-This version of Cephalo, lamm-mit/Cephalo-Idefics2-3x8b-beta, is a Mixture-of-Expert model based on variants and fine-tuned versions of the Idefics-2 model. The basic model architecture is as follows:
+This version of Cephalo, lamm-mit/Cephalo-Idefics2-vision-3x8b-beta, is a Mixture-of-Expert model based on variants and fine-tuned versions of the Idefics-2 model. The basic model architecture is as follows:
 
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/623ce1c6b66fedf374859fe7/b7BK8ZtDzTMsyFDi0wP3w.png)
 
@@ -67,7 +67,7 @@ def count_parameters(model):
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-model_name_moe = f"lamm-mit/Cephalo-Idefics2-3x8b-beta"
+model_name_moe = f"lamm-mit/Cephalo-Idefics2-vision-3x8b-beta"
 config = AutoConfig.from_pretrained(model_name_moe, trust_remote_code=True)
 processor = AutoProcessor.from_pretrained(model_name_moe, trust_remote_code=True) 
 moe_model = AutoModelForCausalLM.from_pretrained(