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-25 19:41:50 +03:00
parent aa30132268
commit bd844331b2
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mochi_preview/t2v_synth_mochi.py
View File

@ -24,10 +24,13 @@ def patched_write_atomic(
write_mode = "w" if isinstance(content, str) else "wb"
with tmp_path.open(write_mode, encoding="utf-8" if encode_utf_8 else None) as f:
f.write(content)
shutil.copy2(src=tmp_path, dst=path)
shutil.copy2(src=tmp_path, dst=path) #to allow overwriting cache files
os.remove(tmp_path)
import torch._inductor.codecache
torch._inductor.codecache.write_atomic = patched_write_atomic
try:
import torch._inductor.codecache
torch._inductor.codecache.write_atomic = patched_write_atomic
except:
pass
import torch
import torch.nn.functional as F