github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Rename backend/lora/ to backend/patches

Browse Source
This commit is contained in:
Ryan Dick
2024-12-13 14:38:50 +00:00
parent 5f41a69665
commit 42f8d6aa11
53 changed files with 123 additions and 121 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
tests/backend/lora/conversions/lora_state_dicts/flux_control_lora_format.py → tests/backend/patches/conversions/lora_state_dicts/flux_control_lora_format.py
View File

0
tests/backend/lora/conversions/lora_state_dicts/flux_lora_diffusers_format.py → tests/backend/patches/conversions/lora_state_dicts/flux_lora_diffusers_format.py
View File

0
tests/backend/lora/conversions/lora_state_dicts/flux_lora_diffusers_no_proj_mlp_format.py → tests/backend/patches/conversions/lora_state_dicts/flux_lora_diffusers_no_proj_mlp_format.py
View File

0
tests/backend/lora/conversions/lora_state_dicts/flux_lora_kohya_format.py → tests/backend/patches/conversions/lora_state_dicts/flux_lora_kohya_format.py
View File

0
tests/backend/lora/conversions/lora_state_dicts/flux_lora_kohya_with_te1_format.py → tests/backend/patches/conversions/lora_state_dicts/flux_lora_kohya_with_te1_format.py
View File

0
tests/backend/lora/conversions/lora_state_dicts/utils.py → tests/backend/patches/conversions/lora_state_dicts/utils.py
View File

10
tests/backend/lora/conversions/test_flux_control_lora_conversion_utils.py → tests/backend/patches/conversions/test_flux_control_lora_conversion_utils.py
View File

@@ -1,18 +1,18 @@
import pytest
import torch
from invokeai.backend.lora.conversions.flux_control_lora_utils import (
from invokeai.backend.patches.conversions.flux_control_lora_utils import (
is_state_dict_likely_flux_control,
lora_model_from_flux_control_state_dict,
)
from invokeai.backend.lora.conversions.flux_lora_constants import FLUX_LORA_TRANSFORMER_PREFIX
from tests.backend.lora.conversions.lora_state_dicts.flux_control_lora_format import (
from invokeai.backend.patches.conversions.flux_lora_constants import FLUX_LORA_TRANSFORMER_PREFIX
from tests.backend.patches.conversions.lora_state_dicts.flux_control_lora_format import (
state_dict_keys as flux_control_lora_state_dict_keys,
)
from tests.backend.lora.conversions.lora_state_dicts.flux_lora_diffusers_format import (
from tests.backend.patches.conversions.lora_state_dicts.flux_lora_diffusers_format import (
state_dict_keys as flux_diffusers_state_dict_keys,
)
from tests.backend.lora.conversions.lora_state_dicts.utils import keys_to_mock_state_dict
from tests.backend.patches.conversions.lora_state_dicts.utils import keys_to_mock_state_dict
@pytest.mark.parametrize("sd_keys", [flux_control_lora_state_dict_keys])

12
tests/backend/lora/conversions/test_flux_diffusers_lora_conversion_utils.py → tests/backend/patches/conversions/test_flux_diffusers_lora_conversion_utils.py
View File

@@ -1,21 +1,21 @@
import pytest
import torch
from invokeai.backend.lora.conversions.flux_diffusers_lora_conversion_utils import (
from invokeai.backend.patches.conversions.flux_diffusers_lora_conversion_utils import (
is_state_dict_likely_in_flux_diffusers_format,
lora_model_from_flux_diffusers_state_dict,
)
from invokeai.backend.lora.conversions.flux_lora_constants import FLUX_LORA_TRANSFORMER_PREFIX
from tests.backend.lora.conversions.lora_state_dicts.flux_lora_diffusers_format import (
from invokeai.backend.patches.conversions.flux_lora_constants import FLUX_LORA_TRANSFORMER_PREFIX
from tests.backend.patches.conversions.lora_state_dicts.flux_lora_diffusers_format import (
state_dict_keys as flux_diffusers_state_dict_keys,
)
from tests.backend.lora.conversions.lora_state_dicts.flux_lora_diffusers_no_proj_mlp_format import (
from tests.backend.patches.conversions.lora_state_dicts.flux_lora_diffusers_no_proj_mlp_format import (
state_dict_keys as flux_diffusers_no_proj_mlp_state_dict_keys,
)
from tests.backend.lora.conversions.lora_state_dicts.flux_lora_kohya_format import (
from tests.backend.patches.conversions.lora_state_dicts.flux_lora_kohya_format import (
state_dict_keys as flux_kohya_state_dict_keys,
)
from tests.backend.lora.conversions.lora_state_dicts.utils import keys_to_mock_state_dict
from tests.backend.patches.conversions.lora_state_dicts.utils import keys_to_mock_state_dict
@pytest.mark.parametrize("sd_keys", [flux_diffusers_state_dict_keys, flux_diffusers_no_proj_mlp_state_dict_keys])

12
tests/backend/lora/conversions/test_flux_kohya_lora_conversion_utils.py → tests/backend/patches/conversions/test_flux_kohya_lora_conversion_utils.py
View File

@@ -4,22 +4,22 @@ import torch
from invokeai.backend.flux.model import Flux
from invokeai.backend.flux.util import params
from invokeai.backend.lora.conversions.flux_kohya_lora_conversion_utils import (
from invokeai.backend.patches.conversions.flux_kohya_lora_conversion_utils import (
_convert_flux_transformer_kohya_state_dict_to_invoke_format,
is_state_dict_likely_in_flux_kohya_format,
lora_model_from_flux_kohya_state_dict,
)
from invokeai.backend.lora.conversions.flux_lora_constants import FLUX_LORA_CLIP_PREFIX, FLUX_LORA_TRANSFORMER_PREFIX
from tests.backend.lora.conversions.lora_state_dicts.flux_lora_diffusers_format import (
from invokeai.backend.patches.conversions.flux_lora_constants import FLUX_LORA_CLIP_PREFIX, FLUX_LORA_TRANSFORMER_PREFIX
from tests.backend.patches.conversions.lora_state_dicts.flux_lora_diffusers_format import (
state_dict_keys as flux_diffusers_state_dict_keys,
)
from tests.backend.lora.conversions.lora_state_dicts.flux_lora_kohya_format import (
from tests.backend.patches.conversions.lora_state_dicts.flux_lora_kohya_format import (
state_dict_keys as flux_kohya_state_dict_keys,
)
from tests.backend.lora.conversions.lora_state_dicts.flux_lora_kohya_with_te1_format import (
from tests.backend.patches.conversions.lora_state_dicts.flux_lora_kohya_with_te1_format import (
state_dict_keys as flux_kohya_te1_state_dict_keys,
)
from tests.backend.lora.conversions.lora_state_dicts.utils import keys_to_mock_state_dict
from tests.backend.patches.conversions.lora_state_dicts.utils import keys_to_mock_state_dict
@pytest.mark.parametrize("sd_keys", [flux_kohya_state_dict_keys, flux_kohya_te1_state_dict_keys])

8
tests/backend/lora/sidecar_layers/concatenated_lora/test_concatenated_lora_linear_sidecar_layer.py → tests/backend/patches/sidecar_layers/concatenated_lora/test_concatenated_lora_linear_sidecar_layer.py
View File

@@ -2,12 +2,12 @@ import copy
import torch
from invokeai.backend.lora.layers.concatenated_lora_layer import ConcatenatedLoRALayer
from invokeai.backend.lora.layers.lora_layer import LoRALayer
from invokeai.backend.lora.sidecar_layers.concatenated_lora.concatenated_lora_linear_sidecar_layer import (
from invokeai.backend.patches.layers.concatenated_lora_layer import ConcatenatedLoRALayer
from invokeai.backend.patches.layers.lora_layer import LoRALayer
from invokeai.backend.patches.sidecar_layers.concatenated_lora.concatenated_lora_linear_sidecar_layer import (
ConcatenatedLoRALinearSidecarLayer,
)
from invokeai.backend.lora.sidecar_layers.lora_sidecar_module import LoRASidecarModule
from invokeai.backend.patches.sidecar_layers.lora_sidecar_module import LoRASidecarModule
def test_concatenated_lora_linear_sidecar_layer():

6
tests/backend/lora/sidecar_layers/lora/test_lora_linear_sidecar_layer.py → tests/backend/patches/sidecar_layers/lora/test_lora_linear_sidecar_layer.py
View File

@@ -2,9 +2,9 @@ import copy
import torch
from invokeai.backend.lora.layers.lora_layer import LoRALayer
from invokeai.backend.lora.sidecar_layers.lora.lora_linear_sidecar_layer import LoRALinearSidecarLayer
from invokeai.backend.lora.sidecar_layers.lora_sidecar_module import LoRASidecarModule
from invokeai.backend.patches.layers.lora_layer import LoRALayer
from invokeai.backend.patches.sidecar_layers.lora.lora_linear_sidecar_layer import LoRALinearSidecarLayer
from invokeai.backend.patches.sidecar_layers.lora_sidecar_module import LoRASidecarModule
@torch.no_grad()

6
tests/backend/lora/test_lora_patcher.py → tests/backend/patches/test_lora_patcher.py
View File

@@ -1,9 +1,9 @@
import pytest
import torch
from invokeai.backend.lora.layers.lora_layer import LoRALayer
from invokeai.backend.lora.lora_model_raw import LoRAModelRaw
from invokeai.backend.lora.lora_patcher import LoRAPatcher
from invokeai.backend.patches.layers.lora_layer import LoRALayer
from invokeai.backend.patches.lora_model_raw import LoRAModelRaw
from invokeai.backend.patches.lora_patcher import LoRAPatcher
class DummyModule(torch.nn.Module):