Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

use peft for lora on non-sd models

Browse Source
This commit is contained in:
Vladimir Mandic
2024-08-30 19:22:02 -04:00
parent e3b54cf699
commit ec281d6acb
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extensions-builtin/Lora/network_overrides.py
+5
View File
@@ -25,7 +25,12 @@ force_diffusers = [ # forced always
]
force_models = [ # forced always
'sc',
'sd3',
'kandinsky',
'hunyuandit',
'auraflow',
'f1',
]
force_classes = [ # forced always
modules/modeldata.py
+3 -1
View File
@@ -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: