xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-05-02 07:53:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove prints

Browse Source
This commit is contained in:
kijai 2025-01-12 19:42:19 +02:00
parent 5a15eea036
commit 7364ce7d2d
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nodes/model_optimization_nodes.py
View File

@ -490,7 +490,7 @@ class TorchCompileCosmosModel:
if not self._compiled: if not self._compiled:
try: try:
for name, block in diffusion_model.blocks.items(): for name, block in diffusion_model.blocks.items():
print(f"Compiling block {name}") #print(f"Compiling block {name}")
compiled_block = torch.compile(block, mode=mode, dynamic=dynamic, fullgraph=fullgraph, backend=backend) compiled_block = torch.compile(block, mode=mode, dynamic=dynamic, fullgraph=fullgraph, backend=backend)
m.add_object_patch(f"diffusion_model.blocks.{name}", compiled_block) m.add_object_patch(f"diffusion_model.blocks.{name}", compiled_block)
#diffusion_model.blocks[name] = compiled_block #diffusion_model.blocks[name] = compiled_block
@ -503,7 +503,6 @@ class TorchCompileCosmosModel:
"dynamic": dynamic, "dynamic": dynamic,
} }
setattr(m.model, "compile_settings", compile_settings) setattr(m.model, "compile_settings", compile_settings)
print(model.model.diffusion_model.blocks)
except: except:
raise RuntimeError("Failed to compile model") raise RuntimeError("Failed to compile model")