Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ def predict(text):
|
|
23 |
return_tensors="pt",).to("cuda")
|
24 |
|
25 |
with torch.autocast("cuda"):
|
26 |
-
outputs = model.generate(**inputs, do_sample=True, guidance_scale=3, max_new_tokens=
|
27 |
|
28 |
return (32000, outputs[0][0].cpu().numpy().astype(np.float16)), gr.make_waveform((32000, outputs[0].cpu().numpy().astype(np.float16).ravel()), bars_color=random.choice(COLORS), bar_count=75)
|
29 |
|
|
|
23 |
return_tensors="pt",).to("cuda")
|
24 |
|
25 |
with torch.autocast("cuda"):
|
26 |
+
outputs = model.generate(**inputs, do_sample=True, guidance_scale=3, max_new_tokens=768)
|
27 |
|
28 |
return (32000, outputs[0][0].cpu().numpy().astype(np.float16)), gr.make_waveform((32000, outputs[0].cpu().numpy().astype(np.float16).ravel()), bars_color=random.choice(COLORS), bar_count=75)
|
29 |
|