Spaces:
Running
on
Zero
Running
on
Zero
Added spaces zerogpu
Browse files
xtts.py
CHANGED
@@ -146,7 +146,7 @@ class TTSInputs(BaseModel):
|
|
146 |
top_p: float
|
147 |
|
148 |
# @app.post("/tts")
|
149 |
-
|
150 |
def predict_speech(parsed_input: TTSInputs):
|
151 |
speaker_embedding = torch.tensor(parsed_input.speaker_embedding).unsqueeze(0).unsqueeze(-1)
|
152 |
gpt_cond_latent = torch.tensor(parsed_input.gpt_cond_latent).reshape((-1, 1024)).unsqueeze(0)
|
|
|
146 |
top_p: float
|
147 |
|
148 |
# @app.post("/tts")
|
149 |
+
@spaces.GPU
|
150 |
def predict_speech(parsed_input: TTSInputs):
|
151 |
speaker_embedding = torch.tensor(parsed_input.speaker_embedding).unsqueeze(0).unsqueeze(-1)
|
152 |
gpt_cond_latent = torch.tensor(parsed_input.gpt_cond_latent).reshape((-1, 1024)).unsqueeze(0)
|