mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-22 05:32:12 -03:00
2b67091986
- Introduced a new WorkflowParser class to streamline workflow parsing and manage node mappers. - Added functionality to load external mappers dynamically from a specified directory. - Refactored LoraLoaderMapper and LoraStackerMapper to handle new data formats for loras and trigger words. - Updated recipe routes to utilize the new WorkflowParser for parsing workflows. - Made adjustments to the flux_prompt.json to reflect changes in active states and class types.
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
"""
|
|
Example extension mapper for demonstrating the extension system
|
|
"""
|
|
from typing import Dict, Any
|
|
from ..mappers import NodeMapper
|
|
|
|
class ExampleNodeMapper(NodeMapper):
|
|
"""Example mapper for custom nodes"""
|
|
|
|
def __init__(self):
|
|
super().__init__(
|
|
node_type="ExampleCustomNode",
|
|
inputs_to_track=["param1", "param2", "image"]
|
|
)
|
|
|
|
def transform(self, inputs: Dict) -> Dict:
|
|
"""Transform extracted inputs into the desired output format"""
|
|
result = {}
|
|
|
|
# Extract interesting parameters
|
|
if "param1" in inputs:
|
|
result["example_param1"] = inputs["param1"]
|
|
|
|
if "param2" in inputs:
|
|
result["example_param2"] = inputs["param2"]
|
|
|
|
# You can process the data in any way needed
|
|
return result
|
|
|
|
|
|
class VAEMapperExtension(NodeMapper):
|
|
"""Extension mapper for VAE nodes"""
|
|
|
|
def __init__(self):
|
|
super().__init__(
|
|
node_type="VAELoader",
|
|
inputs_to_track=["vae_name"]
|
|
)
|
|
|
|
def transform(self, inputs: Dict) -> Dict:
|
|
"""Extract VAE information"""
|
|
vae_name = inputs.get("vae_name", "")
|
|
|
|
# Remove path prefix if present
|
|
if "/" in vae_name or "\\" in vae_name:
|
|
# Get just the filename without path or extension
|
|
vae_name = vae_name.replace("\\", "/").split("/")[-1]
|
|
vae_name = vae_name.split(".")[0] # Remove extension
|
|
|
|
return {"vae": vae_name}
|
|
|
|
|
|
# Note: No need to register manually - extensions are automatically registered
|
|
# when the extension system loads this file |