Skip to content
代码片段 群组 项目
提交 6138f920 编辑于 作者: comfyanonymous's avatar comfyanonymous
浏览文件

Use better dtype for the lowvram lora system.

上级 be0726c1
No related branches found
No related tags found
无相关合并请求
......@@ -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:
......
0% 加载中 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册