File size: 463 Bytes
82ea528 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from ..flux.model import inject_flux
from ..flux.layers import inject_blocks
class ConfigureModifiedFluxNode:
@classmethod
def INPUT_TYPES(s):
return {"required": {
"model": ("MODEL",),
}}
RETURN_TYPES = ("MODEL",)
CATEGORY = "fluxtapoz"
FUNCTION = "apply"
def apply(self, model):
inject_flux(model.model.diffusion_model)
inject_blocks(model.model.diffusion_model)
return (model,)
|