Hugging Face's logo

Spaces:
chendl
/
multimodal
Running

App Files Community
Li commited on
Commit
f407227
•
1 Parent(s): ed32995

“update”

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ flamingo, image_processor, tokenizer, vis_embed_size = create_model_and_transfor
27
  use_format_v2 = True,
28
  )
29
 
30
- checkpoint_path = hf_hub_download("chendl/mm", "checkpoint_opt350m.pt")
31
  checkpoint = torch.load(checkpoint_path, map_location="cpu")
32
  model_state_dict = {}
33
  for key in checkpoint.keys():
 
27
  use_format_v2 = True,
28
  )
29
 
30
+ checkpoint_path = hf_hub_download("chendl/mm", "checkpoint_opt350m_v2.pt")
31
  checkpoint = torch.load(checkpoint_path, map_location="cpu")
32
  model_state_dict = {}
33
  for key in checkpoint.keys():