Update modules/lora.py

modules/lora.py

@@ -7,6 +7,7 @@
 import math
 import os
 import torch
+import diffusers
 import modules.safe as _
 from safetensors.torch import load_file
 
@@ -115,7 +116,10 @@ class LoRANetwork(torch.nn.Module):
         else:
             self.text_encoder_loras = create_modules(LoRANetwork.LORA_PREFIX_TEXT_ENCODER, text_encoder, LoRANetwork.TEXT_ENCODER_TARGET_REPLACE_MODULE)
             print(f"Create LoRA for Text Encoder: {len(self.text_encoder_loras)} modules.")
-
+            
+        if diffusers.__version__ >= "0.15.0":
+            LoRANetwork.UNET_TARGET_REPLACE_MODULE = ["Transformer2DModel"]
+    
         self.unet_loras = create_modules(LoRANetwork.LORA_PREFIX_UNET, unet, LoRANetwork.UNET_TARGET_REPLACE_MODULE)
         print(f"Create LoRA for U-Net: {len(self.unet_loras)} modules.")