298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use better dtype for the lowvram lora system.

Browse Source
This commit is contained in:
comfyanonymous 2024-08-19 15:35:25 -04:00
parent be0726c1ed
commit 6138f92084
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_patcher.py
View File

@ -96,7 +96,7 @@ class LowVramPatch:
self.key = key self.key = key
self.model_patcher = model_patcher self.model_patcher = model_patcher
def __call__(self, weight): def __call__(self, weight):
return self.model_patcher.calculate_weight(self.model_patcher.patches[self.key], weight, self.key) return self.model_patcher.calculate_weight(self.model_patcher.patches[self.key], weight, self.key, intermediate_dtype=weight.dtype)
class ModelPatcher: class ModelPatcher: