mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-25 07:05:43 -03:00
Enhance workflow parsing and node mapper registration
- 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.
This commit is contained in:
@@ -12,6 +12,7 @@ from ..services.civitai_client import CivitaiClient
|
||||
from ..services.recipe_scanner import RecipeScanner
|
||||
from ..services.lora_scanner import LoraScanner
|
||||
from ..config import config
|
||||
from ..workflow.parser import WorkflowParser
|
||||
import time # Add this import at the top
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
@@ -22,6 +23,7 @@ class RecipeRoutes:
|
||||
def __init__(self):
|
||||
self.recipe_scanner = RecipeScanner(LoraScanner())
|
||||
self.civitai_client = CivitaiClient()
|
||||
self.parser = WorkflowParser()
|
||||
|
||||
# Pre-warm the cache
|
||||
self._init_cache_task = None
|
||||
@@ -773,9 +775,7 @@ class RecipeRoutes:
|
||||
latest_image_path = image_files[0][0]
|
||||
|
||||
# Parse the workflow to extract generation parameters and loras
|
||||
from ..workflow_params.workflow_parser import parse_workflow
|
||||
# load_extensions=False to avoid loading extensions for now
|
||||
parsed_workflow = parse_workflow(workflow_json, load_extensions=False)
|
||||
parsed_workflow = self.parser.parse_workflow(workflow_json)
|
||||
|
||||
logger.debug(f"Parsed workflow: {parsed_workflow}")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user