Skip to content
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

Specify optional dependencies for shortfin_apps #513

Merged
merged 2 commits into from
Nov 14, 2024

Conversation

marbre
Copy link
Collaborator

@marbre marbre commented Nov 14, 2024

No description provided.

Comment on lines +37 to +42
[project.optional-dependencies]
apps = [
"transformers",
"dataclasses-json",
"pillow",
]
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This will allow us to simplify the setup instructions https://github.com/nod-ai/SHARK-Platform/blob/3e7429544fd81938e2b1c0d8fbb098b1c871ecf7/shortfin/python/shortfin_apps/sd/README.md?plain=1#L6-L15

to just one of these:

pip install shortfin[apps]
pip install shark-ai[apps]

Comment on lines +33 to +35
apps = [
"shortfin[apps]",
]
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We also considered other names for extras (docs here: https://packaging.python.org/en/latest/specifications/dependency-specifiers/#extras).

  • sd or sdxl: these are very specific, but we don't want to have an extra for every model (sd3, llama2, llama3, flux, etc.), unless there is good reason for that. The transformers, dataclasses-json, and pillow dependencies here aren't specific to SDXL either - they are generic ML / data / image libraries
  • examples or demos: these suggest a lower support level than apps
  • apps: this matches the code path and will let us eventually include other deps for other apps. We can later add more specific extras if apps gets too large

Copy link
Contributor

@monorimet monorimet left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@marbre marbre merged commit 7cc1db3 into nod-ai:main Nov 14, 2024
12 of 13 checks passed
@marbre marbre deleted the shortfin-deps branch November 14, 2024 18:00
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