victor HF Staff commited on
Commit
8ee99c8
·
1 Parent(s): a769062

Fix: pin torchaudio==2.7.1 (CUDA 12 compat), use dtype= over torch_dtype=

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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
- torch_dtype=torch.bfloat16,
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