File size: 327 Bytes
45bc5b2 | 1 2 3 4 5 6 7 8 9 | ---
license: mit
---
Model can be reloaded with the following code snippet:
import torch
from huggingface_hub import hf_hub_download
p = hf_hub_download("jeanq1/experiment_pel", filename="model.ckpt", local_dir="./weights")
ckpt = torch.load(p, map_location="cpu", weights_only=False) # use it like any local file (modifié)
|