shenglin.net/ComfyUI-MochiWrapper-2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update t2v_synth_mochi.py

Browse Source
This commit is contained in:
kijai 2024-11-06 17:27:13 +02:00
parent 76956cda50
commit a46643ae5a
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mochi_preview/t2v_synth_mochi.py
View File

@ -308,10 +308,7 @@ class T2VSynthMochiModel:
comfy_pbar = ProgressBar(sample_steps)
if hasattr(self.dit, "cublas_half_matmul") and self.dit.cublas_half_matmul:
autocast_dtype = torch.float16
else:
if self.device.type == "mps":
if (hasattr(self.dit, "cublas_half_matmul") and self.dit.cublas_half_matmul) or self.device.type == "mps":
autocast_dtype = torch.float16
else:
autocast_dtype = torch.bfloat16