import problem

Author: sayakpaul

tests/lora/test_lora_layers_wanvace.py

@@ -21,11 +21,11 @@
 import pytest
 import safetensors.torch
 import torch
-from peft.utils import get_peft_model_state_dict
 from PIL import Image
 from transformers import AutoTokenizer, T5EncoderModel
 
 from diffusers import AutoencoderKLWan, FlowMatchEulerDiscreteScheduler, WanVACEPipeline, WanVACETransformer3DModel
+from diffusers.utils.import_utils import is_peft_available
 from diffusers.utils.testing_utils import (
     floats_tensor,
     require_peft_backend,
@@ -35,6 +35,9 @@
 )
 
 
+if is_peft_available():
+    from peft.utils import get_peft_model_state_dict
+
 sys.path.append(".")
 
 from utils import PeftLoraLoaderMixinTests  # noqa: E402