Spaces:
Running on Zero
Running on Zero
Fix: pin torchaudio==2.7.1 (CUDA 12 compat), use dtype= over torch_dtype=
Browse files- app.py +1 -1
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -36,7 +36,7 @@ logger.info("Loading TTS model: %s", MODEL_ID)
|
|
| 36 |
tts_model = AutoModelForCausalLM.from_pretrained(
|
| 37 |
MODEL_ID,
|
| 38 |
trust_remote_code=True,
|
| 39 |
-
|
| 40 |
low_cpu_mem_usage=True,
|
| 41 |
)
|
| 42 |
tts_model.eval()
|
|
|
|
| 36 |
tts_model = AutoModelForCausalLM.from_pretrained(
|
| 37 |
MODEL_ID,
|
| 38 |
trust_remote_code=True,
|
| 39 |
+
dtype=torch.bfloat16,
|
| 40 |
low_cpu_mem_usage=True,
|
| 41 |
)
|
| 42 |
tts_model.eval()
|
requirements.txt
CHANGED
|
@@ -3,7 +3,7 @@ sentencepiece>=0.1.99
|
|
| 3 |
transformers==4.57.1
|
| 4 |
safetensors>=0.4.3
|
| 5 |
soundfile>=0.13.1
|
| 6 |
-
torchaudio
|
| 7 |
gradio==6.5.1
|
| 8 |
spaces
|
| 9 |
accelerate>=0.30.0
|
|
|
|
| 3 |
transformers==4.57.1
|
| 4 |
safetensors>=0.4.3
|
| 5 |
soundfile>=0.13.1
|
| 6 |
+
torchaudio==2.7.1
|
| 7 |
gradio==6.5.1
|
| 8 |
spaces
|
| 9 |
accelerate>=0.30.0
|