Skip to content

fix: add tensorrt to all dependencies#425

Open
cbrownstein-lambda wants to merge 1 commit intoblack-forest-labs:mainfrom
cbrownstein-lambda:add-tensorrt-all-dependencies
Open

fix: add tensorrt to all dependencies#425
cbrownstein-lambda wants to merge 1 commit intoblack-forest-labs:mainfrom
cbrownstein-lambda:add-tensorrt-all-dependencies

Conversation

@cbrownstein-lambda
Copy link
Copy Markdown
Contributor

Users expect all to truly include all. Also, without this fix, demo_gr.py for example, won't work:

(.venv) ubuntu@129-146-111-239:~/flux$ python3 demo_gr.py --name flux-schnell --share
Traceback (most recent call last):
  File "/home/ubuntu/flux/demo_gr.py", line 12, in <module>
    from flux.cli import SamplingOptions
  File "/home/ubuntu/flux/src/flux/cli.py", line 8, in <module>
    from cuda import cudart
ModuleNotFoundError: No module named 'cuda'

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.

1 participant