xinyun/ComfyUI-MochiWrapper
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-10-24 00:40:52 +03:00
parent bd954ec132
commit 83097a6b63
1 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mochi_preview/t2v_synth_mochi.py
View File

@ -125,8 +125,8 @@ class T2VSynthMochiModel:
self.offload_device = offload_device self.offload_device = offload_device
print("Initializing model...") print("Initializing model...")
model: nn.Module = torch.nn.utils.skip_init( with (init_empty_weights() if is_accelerate_available else nullcontext()):
AsymmDiTJoint, model: nn.Module = AsymmDiTJoint(
depth=48, depth=48,
patch_size=2, patch_size=2,
num_heads=24, num_heads=24,