Skip to content

Conversation

@patrickvonplaten
Copy link
Contributor

@patrickvonplaten patrickvonplaten commented Nov 28, 2022

This PR speeds up the test by correcting the function argument and removes kwargs from all pipelines which are never used

return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1,
**kwargs,
Copy link
Contributor Author

Choose a reason for hiding this comment

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

I think we forgot to remove them when removing a deprecated function arg. They are never used anywhere


prompt = "hey"

output = sd_pipe(prompt, number_of_steps=1, output_type="np")
Copy link
Contributor Author

Choose a reason for hiding this comment

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

wrong naming here made the test super slow

@patrickvonplaten patrickvonplaten changed the title Remove kwargs from call Speed up test and remove kwargs from call Nov 28, 2022
@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented Nov 28, 2022

The documentation is not available anymore as the PR was closed or merged.

@patrickvonplaten patrickvonplaten force-pushed the remove_kwargs_from_call_in_sd branch from 329128b to 3d99aa8 Compare November 28, 2022 11:38
Copy link
Member

@anton-l anton-l left a comment

Choose a reason for hiding this comment

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

LGTM, kwargs are only used in DDPMPipeline in this version 👍

Copy link
Member

@pcuenca pcuenca left a comment

Choose a reason for hiding this comment

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

Thank you! 🙏

@patrickvonplaten patrickvonplaten merged commit 77fc197 into main Nov 28, 2022
@anton-l anton-l deleted the remove_kwargs_from_call_in_sd branch December 9, 2022 14:52
sliard pushed a commit to sliard/diffusers that referenced this pull request Dec 21, 2022
yoonseokjin pushed a commit to yoonseokjin/diffusers that referenced this pull request Dec 25, 2023
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.

5 participants