diff --git a/__init__.py b/__init__.py index 076fdc8..0efd049 100644 --- a/__init__.py +++ b/__init__.py @@ -12,7 +12,7 @@ from .nodes_registry import ( ) from .nodes_registry import NODES_DISPLAY_NAME_PREFIX, camel_case_to_spaces from .prompt_enhancer_nodes import LTXVPromptEnhancer, LTXVPromptEnhancerLoader -from .q8_nodes import LTXVQ8Patch +from .q8_nodes import LTXVQ8LoraModelLoader, LTXVQ8Patch from .recurrent_sampler import LinearOverlapLatentTransition, LTXVRecurrentKSampler from .stg import ( LTXVApplySTG, @@ -40,6 +40,7 @@ NODE_CLASS_MAPPINGS = { "LTXVPromptEnhancer": LTXVPromptEnhancer, "LTXVPromptEnhancerLoader": LTXVPromptEnhancerLoader, "LTXQ8Patch": LTXVQ8Patch, + "LTXVQ8LoraModelLoader": LTXVQ8LoraModelLoader, "LTXVRecurrentKSampler": LTXVRecurrentKSampler, "LTXVSelectLatents": LTXVSelectLatents, "LTXVSetVideoLatentNoiseMasks": LTXVSetVideoLatentNoiseMasks, diff --git a/q8_nodes.py b/q8_nodes.py index 748b506..19d598b 100644 --- a/q8_nodes.py +++ b/q8_nodes.py @@ -68,8 +68,8 @@ def idendity_quant_fn(x, t): return x.to(dtype=t) -@comfy_node(name="LTXV Q8 Lora Model Loader") -class LoraLoaderQ8ModelOnly: +@comfy_node(name="LTXVQ8LoraModelLoader") +class LTXVQ8LoraModelLoader: def __init__(self): self.loaded_lora = None @@ -87,6 +87,7 @@ class LoraLoaderQ8ModelOnly: } RETURN_TYPES = ("MODEL",) + CATEGORY = "lightricks/LTXV" FUNCTION = "load_lora_model_only" def load_lora(self, model, lora_name, strength_model):