Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -140,7 +140,8 @@ def flex_seq(input_seq, input_file):
|
|
| 140 |
weights_path = get_weights_path(repo_id, file_weights)
|
| 141 |
|
| 142 |
# Load weights
|
| 143 |
-
state_dict = torch.load(weights_path, map_location=config['inference_args']['device'])
|
|
|
|
| 144 |
model.eval()
|
| 145 |
|
| 146 |
data_to_collate = []
|
|
|
|
| 140 |
weights_path = get_weights_path(repo_id, file_weights)
|
| 141 |
|
| 142 |
# Load weights
|
| 143 |
+
state_dict = torch.load(weights_path, map_location=config['inference_args']['device'])
|
| 144 |
+
model.load_state_dict(state_dict, strict=False)
|
| 145 |
model.eval()
|
| 146 |
|
| 147 |
data_to_collate = []
|