Spaces:
Running
on
Zero
Running
on
Zero
Update device setup
Browse files- app.py +2 -0
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -51,6 +51,8 @@ def synthesize_fn(prompt_audio_path, prompt_text, target_text):
|
|
| 51 |
speech_generator.mimi.to("cuda")
|
| 52 |
speech_generator.spk_enc.to("cuda")
|
| 53 |
speech_generator.aligner.aligner.to("cuda")
|
|
|
|
|
|
|
| 54 |
|
| 55 |
if not prompt_audio_path or not target_text:
|
| 56 |
return None
|
|
|
|
| 51 |
speech_generator.mimi.to("cuda")
|
| 52 |
speech_generator.spk_enc.to("cuda")
|
| 53 |
speech_generator.aligner.aligner.to("cuda")
|
| 54 |
+
speech_generator.aligner.device = "cuda"
|
| 55 |
+
speech_generator.device = "cuda"
|
| 56 |
|
| 57 |
if not prompt_audio_path or not target_text:
|
| 58 |
return None
|
requirements.txt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
-
voxtream==0.1.
|
| 2 |
gradio_client==1.3.0
|
| 3 |
pydantic==2.10.6
|
|
|
|
| 1 |
+
voxtream==0.1.2
|
| 2 |
gradio_client==1.3.0
|
| 3 |
pydantic==2.10.6
|