From ec281d6acb4f28120c1b5c058e72cbe9bd6a032d Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Fri, 30 Aug 2024 19:22:02 -0400
Subject: [PATCH] use peft for lora on non-sd models

---
 extensions-builtin/Lora/network_overrides.py | 5 +++++
 modules/modeldata.py                         | 4 +++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/extensions-builtin/Lora/network_overrides.py b/extensions-builtin/Lora/network_overrides.py
index 14018fb70..9123d0039 100644
--- a/extensions-builtin/Lora/network_overrides.py
+++ b/extensions-builtin/Lora/network_overrides.py
@@ -25,7 +25,12 @@ force_diffusers = [ # forced always
 ]
 
 force_models = [ # forced always
+    'sc',
     'sd3',
+    'kandinsky',
+    'hunyuandit',
+    'auraflow',
+    'f1',
 ]
 
 force_classes = [ # forced always
diff --git a/modules/modeldata.py b/modules/modeldata.py
index b114b26b3..f002ca7bd 100644
--- a/modules/modeldata.py
+++ b/modules/modeldata.py
@@ -94,13 +94,15 @@ class Shared(sys.modules[__name__].__class__):
             elif "InstaFlowPipeline" in self.sd_model.__class__.__name__:
                 model_type = 'sd' # instaflow is compatible with sd
             elif "AnimateDiffPipeline" in self.sd_model.__class__.__name__:
-                model_type = 'sd' # sd is compatible with sd
+                model_type = 'sd' # animatediff is compatible with sd
             elif "Kandinsky" in self.sd_model.__class__.__name__:
                 model_type = 'kandinsky'
             elif "HunyuanDiT" in self.sd_model.__class__.__name__:
                 model_type = 'hunyuandit'
             elif "Cascade" in self.sd_model.__class__.__name__:
                 model_type = 'sc'
+            elif "AuraFlow" in self.sd_model.__class__.__name__:
+                model_type = 'auraflow'
             elif "FluxPipeline" in self.sd_model.__class__.__name__:
                 model_type = 'f1'
             else: