Bai-YT commited on
Commit
83cac0f
1 Parent(s): b095b9d

Update run_gradio.py

Browse files
Files changed (1) hide show
  1. run_gradio.py +1 -3
run_gradio.py CHANGED
@@ -48,9 +48,7 @@ def generate(prompt: str, seed: str = '', cfg_weight: float = 4.):
48
  pass
49
 
50
  with torch.no_grad():
51
- with torch.autocast(
52
- device_type="cuda", dtype=torch.bfloat16, enabled=torch.cuda.is_available()
53
- ):
54
  wav = consistencytta(
55
  [prompt], num_steps=1, cfg_scale_input=cfg_weight, cfg_scale_post=1., sr=sr
56
  )
 
48
  pass
49
 
50
  with torch.no_grad():
51
+ with torch.autocast(device_type="cuda", dtype=torch.bfloat16):
 
 
52
  wav = consistencytta(
53
  [prompt], num_steps=1, cfg_scale_input=cfg_weight, cfg_scale_post=1., sr=sr
54
  )