Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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:
|