Pishtiwan MoradiandClaude Sonnet 4.5
92e062ff34
Added models: - Wan2.2 diffusion models: T2V, Animate 14B, T2V Low Noise - Wan2.2 LoRAs: Lightning 4-step, Lightx2v, Relight, Distill - Text encoders: UMT5-XXL (fp8 and bf16 variants) - CLIP Vision H for Wan workflows - InsightFace buffalo_l model for face detection - IP-Adapter Plus SDXL VIT-H - RealESRGAN x4 upscaler - Wan 2.1 VAE Updated file permissions to executable for all Python files and documentation. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
39 lines
2.0 KiB
Python
Executable File
39 lines
2.0 KiB
Python
Executable File
from .model_downloader import NODE_CLASS_MAPPINGS as MODEL_NODES, NODE_DISPLAY_NAME_MAPPINGS as MODEL_NAMES, WEB_DIRECTORY, download_handler
|
|
from .shutdown_monitor import shutdown_status_handler, shutdown_toggle_handler, activity_ping_handler, shutdown_monitor
|
|
from .character_swap_node import NODE_CLASS_MAPPINGS as CHAR_NODES, NODE_DISPLAY_NAME_MAPPINGS as CHAR_NAMES
|
|
from .eye_stabilizer_node import NODE_CLASS_MAPPINGS as EYE_NODES, NODE_DISPLAY_NAME_MAPPINGS as EYE_NAMES
|
|
from .eye_stabilizer_v2_node import NODE_CLASS_MAPPINGS as EYE_V2_NODES, NODE_DISPLAY_NAME_MAPPINGS as EYE_V2_NAMES
|
|
import server
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_CLASS_MAPPINGS.update(MODEL_NODES)
|
|
NODE_CLASS_MAPPINGS.update(CHAR_NODES)
|
|
NODE_CLASS_MAPPINGS.update(EYE_NODES)
|
|
NODE_CLASS_MAPPINGS.update(EYE_V2_NODES)
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(MODEL_NAMES)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CHAR_NAMES)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(EYE_NAMES)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(EYE_V2_NAMES)
|
|
|
|
try:
|
|
if hasattr(server.PromptServer, 'instance') and server.PromptServer.instance is not None:
|
|
routes = server.PromptServer.instance.routes
|
|
routes.post("/model_installer/download")(download_handler)
|
|
routes.get("/pma_utils/shutdown_status")(shutdown_status_handler)
|
|
routes.post("/pma_utils/shutdown_toggle")(shutdown_toggle_handler)
|
|
routes.post("/pma_utils/activity_ping")(activity_ping_handler)
|
|
|
|
shutdown_monitor.set_prompt_server(server.PromptServer.instance)
|
|
|
|
print("[PMA Utils] All routes registered successfully")
|
|
print("[PMA Utils] Shutdown monitor initialized with queue tracking")
|
|
print("[PMA Utils] Character swap nodes loaded")
|
|
print("[PMA Utils] Eye stabilizer node loaded")
|
|
print("[PMA Utils] Eye stabilizer V2 (ethnicity-aware) loaded")
|
|
except Exception as e:
|
|
print(f"[PMA Utils] Warning: Could not register routes immediately: {e}")
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY']
|