V 1.0.0 - Force Lightning Lora by filename

2024-11-11 13:53:24 +01:00
parent 29d8f97988
commit 6b5d02cc4f
1 changed files with 5 additions and 0 deletions
@@ -1148,6 +1148,11 @@ class PrimereCKPTLoader:

        match model_concept:
            case 'Hyper' | 'Lightning':
+                if ckpt_name is not None and model_concept == 'Lightning' and lightning_selector == 'CUSTOM':
+                    is_force_lora = re.findall(r"(FoLo)", ckpt_name)
+                    if len(is_force_lora) > 0:
+                        lightning_selector = 'LORA'
+
                HYPER_LIGHTNING_ORIGINAL_VERSION = utility.getModelType(ckpt_name, 'checkpoints')
                if lightning_selector == 'LORA':
                    Lightning_SDXL_2 = 'https://huggingface.co/ByteDance/SDXL-Lightning/resolve/main/sdxl_lightning_2step_lora.safetensors?download=true'