Spaces:
Running
Running
Pin the weights to the revisions they are fetched at today.
Browse files
app.py
CHANGED
|
@@ -34,7 +34,8 @@ EFFECTS_ORDER = [
|
|
| 34 |
# ---- model loading ----
|
| 35 |
|
| 36 |
def download_and_load(model, ckpt_filename):
|
| 37 |
-
ckpt_path = hf_hub_download(repo_id=HF_REPO, filename=ckpt_filename
|
|
|
|
| 38 |
# checkpoints contain the full pytorch-lightning state dict
|
| 39 |
state_dict = torch.load(ckpt_path, map_location=DEVICE)["state_dict"]
|
| 40 |
model.load_state_dict(state_dict)
|
|
|
|
| 34 |
# ---- model loading ----
|
| 35 |
|
| 36 |
def download_and_load(model, ckpt_filename):
|
| 37 |
+
ckpt_path = hf_hub_download(repo_id=HF_REPO, filename=ckpt_filename,
|
| 38 |
+
revision="c860ca07d4e4fa8d88700c1942c450a98ffd8943")
|
| 39 |
# checkpoints contain the full pytorch-lightning state dict
|
| 40 |
state_dict = torch.load(ckpt_path, map_location=DEVICE)["state_dict"]
|
| 41 |
model.load_state_dict(state_dict)
|