Skip to content

Commit ecda350

Browse files
committed
fix
1 parent 1a5e634 commit ecda350

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/models/autoencoders/test_models_vae.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,6 @@
2121
from datasets import load_dataset
2222
from parameterized import parameterized
2323

24-
25-
if is_peft_available():
26-
from peft import LoraConfig
27-
2824
from diffusers import (
2925
AsymmetricAutoencoderKL,
3026
AutoencoderKL,
@@ -42,6 +38,7 @@
4238
floats_tensor,
4339
is_peft_available,
4440
load_hf_numpy,
41+
require_peft_backend,
4542
require_torch_accelerator,
4643
require_torch_accelerator_with_fp16,
4744
require_torch_gpu,
@@ -55,6 +52,10 @@
5552
from ..test_modeling_common import ModelTesterMixin, UNetTesterMixin
5653

5754

55+
if is_peft_available():
56+
from peft import LoraConfig
57+
58+
5859
enable_full_determinism()
5960

6061

0 commit comments

Comments
 (0)