Spaces:
Running on Zero
Running on Zero
Commit ·
e997409
1
Parent(s): ea0a289
Update app
Browse files
app.py
CHANGED
|
@@ -559,7 +559,7 @@ def load_checkpoint(path, model):
|
|
| 559 |
new_s[k.replace('module.', '')] = v
|
| 560 |
model.load_state_dict(new_s)
|
| 561 |
|
| 562 |
-
model.to(device)
|
| 563 |
|
| 564 |
print("Loaded checkpoint from: {}".format(path))
|
| 565 |
|
|
|
|
| 559 |
new_s[k.replace('module.', '')] = v
|
| 560 |
model.load_state_dict(new_s)
|
| 561 |
|
| 562 |
+
# model.to(device)
|
| 563 |
|
| 564 |
print("Loaded checkpoint from: {}".format(path))
|
| 565 |
|