jeanq1 commited on
Commit
45bc5b2
·
verified ·
1 Parent(s): f772d80

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -3
README.md CHANGED
@@ -1,3 +1,8 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ ---
4
+ Model can be reloaded with the following code snippet:
5
+ import torch
6
+ from huggingface_hub import hf_hub_download
7
+ p = hf_hub_download("jeanq1/experiment_pel", filename="model.ckpt", local_dir="./weights")
8
+ ckpt = torch.load(p, map_location="cpu", weights_only=False) # use it like any local file (modifié)