File size: 435 Bytes
e4ab0d4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | from ..modules.ltx_model import inject_model
class ModifyLTXModelNode:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"model": ("MODEL",),
}
}
RETURN_TYPES = ("MODEL",)
CATEGORY = "ltxtricks"
FUNCTION = "modify"
def modify(self, model):
model.model.diffusion_model = inject_model(model.model.diffusion_model)
return (model,)
|