-
Notifications
You must be signed in to change notification settings - Fork 6.5k
[Community pipeline] Add github mechanism #1680
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
Conversation
|
The documentation is not available anymore as the PR was closed or merged. |
pcuenca
left a comment
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.
Very cool!
|
|
||
| def test_load_custom_github(self): | ||
| pipeline = DiffusionPipeline.from_pretrained( | ||
| "google/ddpm-cifar10-32", custom_pipeline="one_step_unet", custom_revision="0.10.2" |
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.
How does this test verify that the pipeline came from 0.10.2? We could, for example:
- Download from a version where the pipeline did not exist (should fail).
- Download from the first version where it does (succeeds).
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.
Very good point! Will adapt the test :-)
Co-authored-by: Pedro Cuenca <[email protected]>
* [Community pipeline] Add github mechanism * better * Apply suggestions from code review Co-authored-by: Pedro Cuenca <[email protected]> * adapt Co-authored-by: Pedro Cuenca <[email protected]>
* [Community pipeline] Add github mechanism * better * Apply suggestions from code review Co-authored-by: Pedro Cuenca <[email protected]> * adapt Co-authored-by: Pedro Cuenca <[email protected]>
Community pipelines can now be versioned with a
custom_revisioninput tofrom_pretrained. If nothing is specified it will default to what is currently installed locally.