rohansiva commited on
Commit
5510fc8
·
verified ·
1 Parent(s): 19773c4

Update model card for renamed checkpoint file

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -28,7 +28,7 @@ import torch
28
  import open_clip
29
 
30
  model, _, preprocess = open_clip.create_model_and_transforms("ViT-B-32", pretrained="openai")
31
- ckpt = torch.load("new_balanced_clip_finetuned.pt", map_location="cpu")
32
  state_dict = ckpt["model_state_dict"] if "model_state_dict" in ckpt else ckpt
33
  model.load_state_dict(state_dict)
34
  model.eval()
 
28
  import open_clip
29
 
30
  model, _, preprocess = open_clip.create_model_and_transforms("ViT-B-32", pretrained="openai")
31
+ ckpt = torch.load("A4D_model.pt", map_location="cpu")
32
  state_dict = ckpt["model_state_dict"] if "model_state_dict" in ckpt else ckpt
33
  model.load_state_dict(state_dict)
34
  model.eval()