diff --git a/comfy/weight_adapter/base.py b/comfy/weight_adapter/base.py index cbc0c5260..105cc3007 100644 --- a/comfy/weight_adapter/base.py +++ b/comfy/weight_adapter/base.py @@ -114,4 +114,4 @@ def tucker_weight_from_conv(up, down, mid): def tucker_weight(wa, wb, t): temp = torch.einsum("i j ..., j r -> i r ...", t, wb) - return torch.einsum("i j ..., i r -> r j ...", temp, wa) \ No newline at end of file + return torch.einsum("i j ..., i r -> r j ...", temp, wa) diff --git a/comfy/weight_adapter/lora.py b/comfy/weight_adapter/lora.py index 6cb5e98bd..062511b93 100644 --- a/comfy/weight_adapter/lora.py +++ b/comfy/weight_adapter/lora.py @@ -4,10 +4,10 @@ from typing import Optional import torch import comfy.model_management from .base import ( - WeightAdapterBase, - WeightAdapterTrainBase, - weight_decompose, - pad_tensor_to_shape, + WeightAdapterBase, + WeightAdapterTrainBase, + weight_decompose, + pad_tensor_to_shape, tucker_weight_from_conv, )