Update README.md

README.md

@@ -135,7 +135,7 @@ import os
 import shutil
 
 # Repository details
-repo_id = "lamm-mit/Cephalo-Idefics2-3x8b-beta"
+repo_id = "lamm-mit/Cephalo-Idefics2-vision-3x8b-beta"
 api = HfApi()
 
 # List all files in the repository
@@ -177,7 +177,7 @@ from Idefics2_MoE.moe_idefics2 import *
 
 DEVICE='cuda'
 
-model_id_1='lamm-mit/Cephalo-Idefics-2-vision-8b-beta'
+model_id_1='lamm-mit/Cephalo-Idefics2-vision-3x8b-beta'
 
 model_1 = Idefics2ForConditionalGeneration.from_pretrained( model_id_1,
                                                             torch_dtype=torch.bfloat16, #if your GPU allows