Spaces:
Running
on
Zero
Running
on
Zero
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -41,7 +41,7 @@ def load_model_from_hf(repo_id: str = 'jordand/echo-tts-base', device: str = 'cu
|
|
| 41 |
# Now move to device
|
| 42 |
state = {k: v.to(device=device) for k, v in state.items()}
|
| 43 |
|
| 44 |
-
model.load_state_dict(state, strict=
|
| 45 |
model = model.eval()
|
| 46 |
|
| 47 |
if compile:
|
|
|
|
| 41 |
# Now move to device
|
| 42 |
state = {k: v.to(device=device) for k, v in state.items()}
|
| 43 |
|
| 44 |
+
model.load_state_dict(state, strict=False, assign=True)
|
| 45 |
model = model.eval()
|
| 46 |
|
| 47 |
if compile:
|