Spaces:
Running
Running
playback optimization
Browse files
app.py
CHANGED
|
@@ -114,6 +114,9 @@ def generate_samples(model_name, num_samples, num_steps, duration=32768):
|
|
| 114 |
ckpt_path = models[model_name]
|
| 115 |
load_checkpoint(model, ckpt_path)
|
| 116 |
|
|
|
|
|
|
|
|
|
|
| 117 |
with torch.no_grad():
|
| 118 |
all_samples = torch.zeros(2, 0) # initialize all samples
|
| 119 |
for i in range(num_samples):
|
|
|
|
| 114 |
ckpt_path = models[model_name]
|
| 115 |
load_checkpoint(model, ckpt_path)
|
| 116 |
|
| 117 |
+
if num_samples > 1:
|
| 118 |
+
duration = duration / 2
|
| 119 |
+
|
| 120 |
with torch.no_grad():
|
| 121 |
all_samples = torch.zeros(2, 0) # initialize all samples
|
| 122 |
for i in range(num_samples):
|