lora load cache state_dict

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-07-01 18:17:10 +02:00
parent e1ac9650b7
commit 22d01c7e6e
4 changed files with 48 additions and 10 deletions
+1 -1
View File
@@ -92,7 +92,7 @@ def guess_by_name(fn, current_guess):
new_guess = 'ZetaChroma'
elif 'chroma' in fn.lower() and 'xl' not in fn.lower():
new_guess = 'Chroma'
elif 'flux.2' in fn.lower() and 'klein' in fn.lower():
elif ('flux.2' in fn.lower() or 'f2' in fn.lower()) and 'klein' in fn.lower():
new_guess = 'FLUX2Klein'
elif 'flux.2' in fn.lower():
new_guess = 'FLUX2'