fix_sdxl_light_test_scrip (#735)

siliconflow · Mar 18, 2024 · cef3251 · cef3251
1 parent 293d7cb
commit cef3251
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/onediff_diffusers_extensions/examples/text_to_image_sdxl_light.py b/onediff_diffusers_extensions/examples/text_to_image_sdxl_light.py
@@ -7,7 +7,11 @@
 from diffusers import StableDiffusionXLPipeline
 from onediffx import compile_pipe, compiler_config, save_pipe, load_pipe
 from huggingface_hub import hf_hub_download
-from diffusers.utils import USE_PEFT_BACKEND
+
+try:
+    USE_PEFT_BACKEND = diffusers.utils.USE_PEFT_BACKEND
+except Exception as e:
+    USE_PEFT_BACKEND = False
 
 parser = argparse.ArgumentParser()
 parser.add_argument(