FearL0rd
838052e0ff
Added VRAM purge toggles to INPUT_TYPES: purge_cache: Clears CUDA cache during cleanup (default: True) purge_models: Aggressively unloads all models from VRAM using comfy.model_management.unload_all_models() (default: False) Updated setup_parallel method signature to accept purge_cache and purge_models parameters Store purge preferences on the model: target_model._parallel_purge_cache and target_model._parallel_purge_models Enhanced cleanup_parallel_model function to: Read the purge flags from the model attributes Conditionally run comfy.model_management.unload_all_models() if purge_models is True Conditionally clear CUDA cache and soft empty cache based on purge_cache setting Multi-GPU cache clearing (iterates through all CUDA devices) The purge settings persist with the model and are applied when the parallel processing resources are cleaned up (either when the node is re-run with new settings or when the model is garbage collected).
15 lines
1.0 KiB
TOML
15 lines
1.0 KiB
TOML
# pyproject.toml
|
|
[project]
|
|
name = "ComfyUI-ParallelAnything" # Unique identifier for your node. Immutable after creation.
|
|
description = "This suite of nodes unlocks high-performance parallel processing in ComfyUI by utilizing **Model Replication**. Unlike standard offloading which moves a single model instance between devices, these nodes create independent replicas of the model on each selected GPU/CPU, allowing for true simultaneous batch processing."
|
|
version = "1.0.4" # Custom Node version. Must be semantically versioned.
|
|
license = { file = "LICENSE.txt" }
|
|
dependencies = [] # Filled in from requirements.txt
|
|
|
|
[project.urls]
|
|
Repository = "https://github.com/FearL0rd/ComfyUI-ParallelAnything"
|
|
|
|
[tool.comfy]
|
|
PublisherId = "fearl0rd" # TODO (fill in Publisher ID from Comfy Registry Website).
|
|
DisplayName = "ComfyUI-ParallelAnything" # Display name for the Custom Node. Can be changed later.
|
|
Icon = "https://raw.githubusercontent.com/FearL0rd/ComfyUI-ParallelAnything/refs/heads/main/icon.png" # SVG, PNG, JPG or GIF (MAX. 800x400px) |