32 lines
794 B
Python
32 lines
794 B
Python
class DITConfig:
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"optional": {
|
|
"prefix": ("STRING", {
|
|
"default": ""
|
|
}),
|
|
"quant_config": ("STRING", {
|
|
"default": ""
|
|
}),
|
|
}
|
|
}
|
|
|
|
@classmethod
|
|
def VALIDATE_INPUTS(cls, **kwargs):
|
|
return True
|
|
|
|
RETURN_TYPES = ("DIT_CONFIG", )
|
|
RETURN_NAMES = ("dit_config", )
|
|
FUNCTION = "set_args"
|
|
CATEGORY = "fastvideo"
|
|
|
|
def set_args(self, prefix, quant_config):
|
|
raw_args = {"prefix": prefix, "quant_config": quant_config}
|
|
|
|
# Filter out keys where value is -99999
|
|
args = {k: v for k, v in raw_args.items() if str(int(v)) != str(-99999)}
|
|
|
|
return (args, )
|