jaidaken
f09734b0ee
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
47 lines
2.0 KiB
Python
47 lines
2.0 KiB
Python
from .constants import get_category, get_name
|
|
from nodes import LoraLoader
|
|
import folder_paths
|
|
|
|
|
|
class RgthreeLoraLoaderStack:
|
|
|
|
NAME = get_name('Lora Loader Stack')
|
|
CATEGORY = get_category()
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls): # pylint: disable = invalid-name, missing-function-docstring
|
|
return {
|
|
"required": {
|
|
"model": ("MODEL",),
|
|
"clip": ("CLIP", ),
|
|
|
|
"lora_01": (['None'] + folder_paths.get_filename_list("loras"), ),
|
|
"strength_01":("FLOAT", {"default": 1.0, "min": -10.0, "max": 10.0, "step": 0.01}),
|
|
|
|
"lora_02": (['None'] + folder_paths.get_filename_list("loras"), ),
|
|
"strength_02":("FLOAT", {"default": 1.0, "min": -10.0, "max": 10.0, "step": 0.01}),
|
|
|
|
"lora_03": (['None'] + folder_paths.get_filename_list("loras"), ),
|
|
"strength_03":("FLOAT", {"default": 1.0, "min": -10.0, "max": 10.0, "step": 0.01}),
|
|
|
|
"lora_04": (['None'] + folder_paths.get_filename_list("loras"), ),
|
|
"strength_04":("FLOAT", {"default": 1.0, "min": -10.0, "max": 10.0, "step": 0.01}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("MODEL", "CLIP")
|
|
FUNCTION = "load_lora"
|
|
|
|
def load_lora(self, model, clip, lora_01, strength_01, lora_02, strength_02, lora_03, strength_03, lora_04, strength_04):
|
|
if lora_01 != "None" and strength_01 != 0:
|
|
model, clip = LoraLoader().load_lora(model, clip, lora_01, strength_01, strength_01)
|
|
if lora_02 != "None" and strength_02 != 0:
|
|
model, clip = LoraLoader().load_lora(model, clip, lora_02, strength_02, strength_02)
|
|
if lora_03 != "None" and strength_03 != 0:
|
|
model, clip = LoraLoader().load_lora(model, clip, lora_03, strength_03, strength_03)
|
|
if lora_04 != "None" and strength_04 != 0:
|
|
model, clip = LoraLoader().load_lora(model, clip, lora_04, strength_04, strength_04)
|
|
|
|
return (model, clip)
|
|
|