lamm-mit
/

Cephalo-Idefics2-vision-3x8b-beta

Model card Files Files and versions Community

mjbuehler commited on Jun 9

Commit

be96f5f

•

1 Parent(s): f70961d

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -135,7 +135,7 @@ import os
 import shutil
 # Repository details
-repo_id = "lamm-mit/Cephalo-Idefics2-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_1 = Idefics2ForConditionalGeneration.from_pretrained( model_id_1,
                                                             torch_dtype=torch.bfloat16, #if your GPU allows

 import shutil
 # Repository details
+repo_id = "lamm-mit/Cephalo-Idefics2-vision-3x8b-beta"
 api = HfApi()
 # List all files in the repository
 DEVICE='cuda'
+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