Dia-1.6B

Running on T4

buttercrab commited on 4 days ago

Commit

596ef1a

1 Parent(s): 563368d

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ from dia.model import Dia
 print("Loading Nari model...")
 try:
     # Use the function from inference.py
-    model = Dia.from_pretrained("nari-labs/Dia-1.6B", compute_dtype="bfloat16")
 except Exception as e:
     print(f"Error loading Nari model: {e}")
     raise

 print("Loading Nari model...")
 try:
     # Use the function from inference.py
+    model = Dia.from_pretrained("nari-labs/Dia-1.6B", compute_dtype="float32")
 except Exception as e:
     print(f"Error loading Nari model: {e}")
     raise