wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-30 13:33:42 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Merge 56e9de4f9f into b353a7c863

Browse Source
This commit is contained in:
Soulreaver90 2026-03-28 04:39:45 +00:00 committed by GitHub
commit aa736fefc2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_patcher.py
View File

@ -122,6 +122,11 @@ class LowVramPatch:
self.set_func = set_func
def __call__(self, weight):
if weight.dtype == torch.float8_e4m3fn or weight.dtype == torch.float8_e5m2:
temp_weight = weight.to(torch.bfloat16)
patched_weight = comfy.lora.calculate_weight(self.patches[self.key], temp_weight, self.key, intermediate_dtype=torch.bfloat16)
return patched_weight.to(weight.dtype)
return comfy.lora.calculate_weight(self.patches[self.key], weight, self.key, intermediate_dtype=weight.dtype)
LOWVRAM_PATCH_ESTIMATE_MATH_FACTOR = 2