Spaces:
Sleeping
Sleeping
Update run_gradio.py
Browse files- 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 |
)
|