jordand commited on
Commit
0fe8b05
·
verified ·
1 Parent(s): f81c714

Update inference.py

Browse files
Files changed (1) hide show
  1. 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=True, assign=True)
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: