-
Notifications
You must be signed in to change notification settings - Fork 431
add vui code snippet #1522
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
add vui code snippet #1522
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @Deep-unlearning ! Deferring to @Vaibhavs10 for code snippet review.
prettyLabel: "VUI", | ||
repoName: "VUI", | ||
repoUrl: "https://github.com/vui-ai/vui", | ||
countDownloads: `path_extension:"pt"`, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Would be really good to nudge them to split their repo into several ones (better to track downloads on individual models).
(highly recommended but not an obligation)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
yes, i am pushing for it!
from vui.inference import render | ||
from vui.model import Vui, | ||
|
||
model = Vui.from_pretrained().cuda() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
typically here if repo is split in several models, we could have a better .from_pretrained
snippet. Requires some slight changes in the GH repo as well but worth it IMO
Co-authored-by: Lucain <[email protected]>
Co-authored-by: Lucain <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Good to merge on myside, make sure to fix the CI error before merge
Lint error introduced in #1522. Will merge as soon as lint CI is green
Waiting for this PR to be merged:
https://huggingface.co/fluxions/vui/discussions/1/files