diffusers: AttributeError: 'Linear' object has no attribute 'set_lora_layer'

Describe the bug

I use (ttps://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_lora.py to train lora and raise an error AttributeError: Linear' object has no attribute 'set_lora_layer.

Reproduction

training examples/text_to_image/train_text_to_image_lora.py

Logs

No response

System Info

• diffusers version: 0.24.0.dev0
• Platform: Linux-4.15.0-142-generic-x86_64-with-glibc2.27
• Python version: 3.10.13
• PyTorch version (GPU?): 2.1.0+cu121 (True)
• Huggingface_hub version: 0.19.4
• Transformers version: 4.35.2
• Accelerate version: 0.24.1
• xFormers version: 0.0.22.post7
• Using GPU in script?: <fill in>
• Using distributed or parallel set-up in script?: <fill in>

Who can help?

@sayakpaul @patrickvonplaten