Skip to content

Commit 3b0368a

Browse files
Fix regression. (#11194)
1 parent 935493f commit 3b0368a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

comfy/model_patcher.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,9 @@ def low_vram_patch_estimate_vram(model, key):
139139
if weight is None:
140140
return 0
141141
model_dtype = getattr(model, "manual_cast_dtype", torch.float32)
142+
if model_dtype is None:
143+
model_dtype = weight.dtype
144+
142145
return weight.numel() * model_dtype.itemsize * LOWVRAM_PATCH_ESTIMATE_MATH_FACTOR
143146

144147
def get_key_weight(model, key):

0 commit comments

Comments
 (0)