reach-vb HF Staff commited on
Commit
a8ea6f3
·
1 Parent(s): 77e74a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=512)
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