Devakumar868 commited on
Commit
baf7f5d
·
verified ·
1 Parent(s): 00432e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,10 +58,10 @@ def load_models():
58
  if dia_model is None:
59
  print("Loading Dia TTS model...")
60
  try:
 
61
  dia_model = Dia.from_pretrained(
62
  "nari-labs/Dia-1.6B",
63
- compute_dtype="float16",
64
- torch_dtype=torch.float16
65
  )
66
  print("✅ Dia model loaded successfully!")
67
  except Exception as e:
 
58
  if dia_model is None:
59
  print("Loading Dia TTS model...")
60
  try:
61
+ # FIXED: Remove torch_dtype parameter - only use compute_dtype
62
  dia_model = Dia.from_pretrained(
63
  "nari-labs/Dia-1.6B",
64
+ compute_dtype="float16"
 
65
  )
66
  print("✅ Dia model loaded successfully!")
67
  except Exception as e: