Hugging Face's logo

Spaces:
ntt123
/
diffusion-speech-360h
Runtime error

App Files Community
ntt123 commited on
Commit
94055f6
·
1 Parent(s): 02d623b

use fp32

Browse files
Files changed (1) hide show
  1. sample.py +0 -1
sample.py CHANGED
@@ -179,7 +179,6 @@ def sample(
179
  state_dict = find_model(ckpt_path)
180
  model.load_state_dict(state_dict)
181
  model.eval() # important!
182
- model = model.bfloat16()
183
  model_cache[ckpt_path] = model
184
  else:
185
  model = model_cache[ckpt_path]
 
179
  state_dict = find_model(ckpt_path)
180
  model.load_state_dict(state_dict)
181
  model.eval() # important!
 
182
  model_cache[ckpt_path] = model
183
  else:
184
  model = model_cache[ckpt_path]