Sirshendu Ganguly
056ade0661
- Use librosa instead of torchaudio for mediaStorage - Created dedicated toggle handlers for Image Inference, Video Inference, Upscaler, Audio Inference, Accelerator Options, Bytedance Provider Settings, and OpenAI Provider Settings - Fixed Audio Inference widget ordering (use parameters now appear before their corresponding parameters) - All handlers use the same proven pattern from videoInferenceDimensionsHandler - Improved widget disabling/enabling functionality across all nodes
57 lines
2.6 KiB
Python
57 lines
2.6 KiB
Python
class refiner:
|
|
"""Refiner node for enhancing image quality using refiner models"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"Model": ("RUNWAREMODEL", {
|
|
"tooltip": "Connect a Runware Model From Model Search Node.",
|
|
}),
|
|
"startStep": ("INT", {
|
|
"tooltip": "Represents the step number at which the refinement process begins. The initial model will generate the image up to this step, after which the refiner model takes over to enhance the result.",
|
|
"min": 0,
|
|
"max": 99,
|
|
"default": 0,
|
|
}),
|
|
"startStepPercentage": ("INT", {
|
|
"tooltip": "Represents the percentage of total steps at which the refinement process begins. The initial model will generate the image up to this percentage of steps before the refiner takes over.",
|
|
"min": 0,
|
|
"max": 99,
|
|
"default": 0
|
|
}),
|
|
},
|
|
}
|
|
|
|
DESCRIPTION = "Refiner models help create higher quality image outputs by incorporating specialized models designed to enhance image details and overall coherence. This can be particularly useful when you need results with superior quality, photorealism, or specific aesthetic refinements. (Note that refiner models are only SDXL based)"
|
|
FUNCTION = "refiner"
|
|
RETURN_TYPES = ("RUNWAREREFINER",)
|
|
CATEGORY = "Runware"
|
|
|
|
@classmethod
|
|
def VALIDATE_INPUTS(cls, startStep, startStepPercentage):
|
|
"""Validate that exactly one of startStep or startStepPercentage is provided"""
|
|
if startStep != 0 and startStepPercentage != 0:
|
|
raise Exception("Please provide either startStep or startStepPercentage, not both.")
|
|
elif startStep == 0 and startStepPercentage == 0:
|
|
raise Exception("Please provide either startStep or startStepPercentage.")
|
|
else:
|
|
return True
|
|
|
|
def refiner(self, **kwargs):
|
|
"""Create refiner configuration"""
|
|
runwareModel = kwargs.get("Model")
|
|
startStep = kwargs.get("startStep")
|
|
startStepPercentage = kwargs.get("startStepPercentage")
|
|
|
|
refinerResult = {
|
|
"model": runwareModel,
|
|
}
|
|
|
|
if startStep != 0:
|
|
refinerResult["startStep"] = startStep
|
|
else:
|
|
refinerResult["startStepPercentage"] = startStepPercentage
|
|
|
|
return (refinerResult,)
|