Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -128,7 +128,7 @@ def modelspeech(text,name_model):
|
|
128 |
|
129 |
inputs = tokenizer(text, return_tensors="pt")
|
130 |
model=get_model(name_model)
|
131 |
-
|
132 |
wav=list(_inference_forward_stream(model,input_ids=inputs.input_ids.cuda(),attention_mask=inputs.attention_mask.cuda(),speaker_embeddings= None,is_streaming=False))[0]
|
133 |
# with torch.no_grad():
|
134 |
# wav = model(input_ids=inputs["input_ids"].cuda()).waveform.cpu().numpy().reshape(-1)#.detach()
|
|
|
128 |
|
129 |
inputs = tokenizer(text, return_tensors="pt")
|
130 |
model=get_model(name_model)
|
131 |
+
with torch.no_grad():
|
132 |
wav=list(_inference_forward_stream(model,input_ids=inputs.input_ids.cuda(),attention_mask=inputs.attention_mask.cuda(),speaker_embeddings= None,is_streaming=False))[0]
|
133 |
# with torch.no_grad():
|
134 |
# wav = model(input_ids=inputs["input_ids"].cuda()).waveform.cpu().numpy().reshape(-1)#.detach()
|