Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7):
|
|
40 |
|
41 |
print(f"Sample rate: {sample_rate}, Sample size: {sample_size}")
|
42 |
|
43 |
-
model
|
44 |
print("Model moved to device.")
|
45 |
|
46 |
# Set up text and timing conditioning
|
|
|
40 |
|
41 |
print(f"Sample rate: {sample_rate}, Sample size: {sample_size}")
|
42 |
|
43 |
+
model.to(device)
|
44 |
print("Model moved to device.")
|
45 |
|
46 |
# Set up text and timing conditioning
|