Skip to content

[tests] guard peft imports in test_lora_loader_utils #14434

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
sayakpaul merged 1 commit into from
Aug 10, 2026
+9 −2
Merged

[tests] guard peft imports in test_lora_loader_utils #14434

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 9 additions & 2 deletions tests/lora/test_lora_loader_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,15 +18,21 @@
import pytest
import torch
import torch.nn as nn
from peft import LoraConfig
from peft.tuners.tuners_utils import BaseTunerLayer
from safetensors.torch import save_file

from diffusers.configuration_utils import ConfigMixin
from diffusers.loaders import StableDiffusionLoraLoaderMixin, lora_base
from diffusers.loaders.lora_base import LoraBaseMixin
from diffusers.loaders.peft import PeftAdapterMixin
from diffusers.models.modeling_utils import ModelMixin
from diffusers.utils.import_utils import is_peft_available

from ..testing_utils import require_peft_backend


if is_peft_available():
from peft import LoraConfig
from peft.tuners.tuners_utils import BaseTunerLayer


LORA_KEY = "unet.test.lora_A.weight"
Expand Down Expand Up @@ -109,6 +115,7 @@ def test_local_directory_with_multiple_files_warns_and_uses_first(tmp_path, monk
assert "contains more than one weights file" in caplog.text


@require_peft_backend
def test_unfuse_lora_partial_components_keeps_merged_adapters_in_sync():
"""Regression test for gh-14214.

Expand Down
Loading