WeiChow commited on
Commit
3cb7004
·
verified ·
1 Parent(s): 0e10429

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -10,7 +10,7 @@ model = MoGeModel.from_pretrained("Ruicheng/moge-vitl").to(device)
10
  # -------------------
11
  backbone_state_dict = model.backbone.state_dict()
12
  filtered_state_dict = {k: v for k, v in backbone_state_dict.items() if 'mask_token' not in k}
13
- torch.save(filtered_state_dict, "moge_l_vit.bin")
14
  # -------------------
15
  ```
16
 
 
10
  # -------------------
11
  backbone_state_dict = model.backbone.state_dict()
12
  filtered_state_dict = {k: v for k, v in backbone_state_dict.items() if 'mask_token' not in k}
13
+ torch.save(filtered_state_dict, "pytorch_model.bin")
14
  # -------------------
15
  ```
16