Sirshendu Ganguly
e96fc12fc4
- Add Runware Text to Video and Image to Video nodes - Add FLUX.1 Krea model to all search modules
87 lines
3.5 KiB
Python
87 lines
3.5 KiB
Python
class loraSearch:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"Lora Search": ("STRING", {
|
|
"tooltip": "Search For Specific Lora By Name Or Civit AIR Code (eg: Cyberpunk).",
|
|
}),
|
|
"Model Architecture": ([
|
|
"All",
|
|
"FLUX.1-Schnell",
|
|
"FLUX.1-Dev",
|
|
"FLUX.1 Krea",
|
|
"Pony",
|
|
"SD 1.5",
|
|
"SD 1.5 Hyper",
|
|
"SD 1.5 LCM",
|
|
"SD 3",
|
|
"SDXL 1.0",
|
|
"SDXL 1.0 LCM",
|
|
"SDXL Distilled",
|
|
"SDXL Hyper",
|
|
"SDXL Lightning",
|
|
"SDXL Turbo",
|
|
], {
|
|
"tooltip": "Choose Lora Model Architecture To Filter Results.",
|
|
"default": "All",
|
|
}),
|
|
"LoraType": ([
|
|
"Lora",
|
|
"LyCORIS"
|
|
], {
|
|
"tooltip": "Choose Lora Type To Filter Out The Results.",
|
|
"default": "Lora",
|
|
}),
|
|
"LoraList": ([
|
|
"civitai:58390@62833 (Detail Tweaker LoRA (细节调整LoRA))",
|
|
"civitai:82098@87153 (Add More Details - Detail Enhancer / Tweaker (细节调整) LoRA)",
|
|
"civitai:122359@135867 (Detail Tweaker XL)",
|
|
"civitai:14171@16677 (Cute_girl_mix4)",
|
|
"civitai:13941@16576 (epi_noiseoffset)",
|
|
"civitai:25995@32988 (blindbox/大概是盲盒)",
|
|
], {
|
|
"tooltip": "Lora Results Will Show UP Here So You Could Choose From.",
|
|
"default": "civitai:58390@62833 (Detail Tweaker LoRA (细节调整LoRA))",
|
|
}),
|
|
"weight": ("FLOAT", {
|
|
"tooltip": "Defines the strength or influence of the LoRA model in the generation process.",
|
|
"default": 1.0,
|
|
"min": -4.0,
|
|
"max": 4.0,
|
|
"step": 0.1,
|
|
}),
|
|
"Use Search Value": ("BOOLEAN", {
|
|
"tooltip": "When Enabled, the value you've set in the search input will be used instead.\n\nThis is useful in case the model search API is down or you prefer to set the model manually.",
|
|
"default": False,
|
|
"label_on": "Enabled",
|
|
"label_off": "Disabled",
|
|
}),
|
|
},
|
|
}
|
|
|
|
DESCRIPTION = "Directly Search and Connect Lora's to Runware Inference Nodes In ComfyUI."
|
|
FUNCTION = "loraSearch"
|
|
RETURN_TYPES = ("RUNWARELORA",)
|
|
RETURN_NAMES = ("Runware Lora",)
|
|
CATEGORY = "Runware"
|
|
|
|
@classmethod
|
|
def VALIDATE_INPUTS(cls, LoraList):
|
|
return True
|
|
|
|
def loraSearch(self, **kwargs):
|
|
enableSearchValue = kwargs.get("Use Search Value", False)
|
|
searchInput = kwargs.get("Lora Search")
|
|
lora_weight = kwargs.get("weight")
|
|
|
|
if enableSearchValue:
|
|
modelAIRCode = searchInput
|
|
else:
|
|
crModel = kwargs.get("LoraList")
|
|
modelAIRCode = crModel.split(" ")[0]
|
|
|
|
return ({
|
|
"model": modelAIRCode,
|
|
"weight": round(lora_weight, 2),
|
|
},) |