Update generate.py

generate.py

@@ -142,7 +142,7 @@ def generate(
     use_complex_talk_head=True,
     use_weighted_talk_head=True,
     trust_remote_code=True,
-    torch_dtype=torch.bfloat16,
+    torch_dtype=None,
     **model_kwargs,
 ):
     # Set model attributes