wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-05 19:12:41 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

ops: fix __init__ return

Browse Source
This commit is contained in:
Rattus 2026-01-22 00:02:11 +10:00
parent 4979c075c9
commit 65b9729912
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ops.py
View File

@ -283,7 +283,8 @@ class disable_weight_init:
def __init__(self, in_features, out_features, bias=True, device=None, dtype=None): def __init__(self, in_features, out_features, bias=True, device=None, dtype=None):
if not comfy.model_management.WINDOWS or not enables_dynamic_vram(): if not comfy.model_management.WINDOWS or not enables_dynamic_vram():
return super().__init__(in_features, out_features, bias, device, dtype) super().__init__(in_features, out_features, bias, device, dtype)
return
# Issue is with `torch.empty` still reserving the full memory for the layer. # Issue is with `torch.empty` still reserving the full memory for the layer.
# Windows doesn't over-commit memory so without this, We are momentarily commit # Windows doesn't over-commit memory so without this, We are momentarily commit