File size: 401 Bytes
82ea528 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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, )
|