Update app.py
Browse files
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="
|
| 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()
|