From 0f356a9212cee4d2594ada1d6f9b9563a5e31372 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Mon, 19 Aug 2024 15:35:25 -0400 Subject: [PATCH] Use better dtype for the lowvram lora system. --- comfy/model_patcher.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index 8dafc54a3..51259b559 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -96,7 +96,7 @@ class LowVramPatch: self.key = key self.model_patcher = model_patcher 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: