Skip to content

Fix custom node blocker for multi-GPU inference using multi-processing #8269

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

SolitaryThinker
Copy link

Why these changes are needed

We have created a set of custom ComfyUI nodes around FastVideo, a framework for multi-GPU video generation using sequence parallelism.

This fixes import failures that were occurring when child processes attempted to import modules, since spawned processes get their own Python interpreter and don't inherit the same import context as the parent process.

Custom Node: https://github.com/kevin314/ComfyUI-FastVideo

Changes

Adds the directory containing main.py to Python's module search path to ensure spawned processes from multiprocessing can correctly find and import local modules.

We don't want to have users install a comfyui fork so would like to upstream this fix

Thanks

@bigcat88
Copy link
Contributor

Hello. Did you tried to modify PYTHONPATH environment variable for that, does it helps in your case?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants