rrg92 commited on
Commit
592c43f
·
1 Parent(s): 74b5d82

Added spaces zerogpu

Browse files
Files changed (1) hide show
  1. xtts.py +1 -1
xtts.py CHANGED
@@ -146,7 +146,7 @@ class TTSInputs(BaseModel):
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)
 
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)