FlameF0X commited on
Commit
773fa52
·
verified ·
1 Parent(s): 90bee26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -215,7 +215,7 @@ model = i3CLIPHybridLarge(tokenizer.vocab_size).to(device)
215
 
216
  # Load checkpoint from FlameF0X/i3-CLIP
217
  print("Downloading and loading model weights...")
218
- checkpoint_path = hf_hub_download(repo_id="FlameF0X/i3-CLIP", filename="pytorch_model.bin")
219
  state_dict = torch.load(checkpoint_path, map_location=device)
220
  model.load_state_dict(state_dict, strict=False)
221
  model.eval()
 
215
 
216
  # Load checkpoint from FlameF0X/i3-CLIP
217
  print("Downloading and loading model weights...")
218
+ checkpoint_path = hf_hub_download(repo_id="i3-lab/i3-CLIP", filename="pytorch_model.bin")
219
  state_dict = torch.load(checkpoint_path, map_location=device)
220
  model.load_state_dict(state_dict, strict=False)
221
  model.eval()