mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-22 13:42:12 -03:00
fc4327087b
- Implemented the WanVideo Lora Select node in Python with input handling for low memory loading and LORA syntax processing. - Updated the JavaScript side to register the new node and manage its widget interactions. - Enhanced constants files to include the new node type and its corresponding ID. - Modified existing Lora Loader and Stacker references to accommodate the new node in various workflows and UI components. - Added example workflow JSON for the new node to demonstrate its usage.
49 lines
1.0 KiB
Python
49 lines
1.0 KiB
Python
NSFW_LEVELS = {
|
|
"PG": 1,
|
|
"PG13": 2,
|
|
"R": 4,
|
|
"X": 8,
|
|
"XXX": 16,
|
|
"Blocked": 32, # Probably not actually visible through the API without being logged in on model owner account?
|
|
}
|
|
|
|
# Node type constants
|
|
NODE_TYPES = {
|
|
"Lora Loader (LoraManager)": 1,
|
|
"Lora Stacker (LoraManager)": 2,
|
|
"WanVideo Lora Select (LoraManager)": 3
|
|
}
|
|
|
|
# Default ComfyUI node color when bgcolor is null
|
|
DEFAULT_NODE_COLOR = "#353535"
|
|
|
|
# preview extensions
|
|
PREVIEW_EXTENSIONS = [
|
|
'.webp',
|
|
'.preview.webp',
|
|
'.preview.png',
|
|
'.preview.jpeg',
|
|
'.preview.jpg',
|
|
'.preview.mp4',
|
|
'.png',
|
|
'.jpeg',
|
|
'.jpg',
|
|
'.mp4',
|
|
'.gif',
|
|
'.webm'
|
|
]
|
|
|
|
# Card preview image width
|
|
CARD_PREVIEW_WIDTH = 480
|
|
|
|
# Width for optimized example images
|
|
EXAMPLE_IMAGE_WIDTH = 832
|
|
|
|
# Supported media extensions for example downloads
|
|
SUPPORTED_MEDIA_EXTENSIONS = {
|
|
'images': ['.jpg', '.jpeg', '.png', '.webp', '.gif'],
|
|
'videos': ['.mp4', '.webm']
|
|
}
|
|
|
|
# Valid Lora types
|
|
VALID_LORA_TYPES = ['lora', 'locon', 'dora'] |