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

Clean FID eval for SDXL #86

Merged
merged 10 commits into from
Nov 7, 2023
Merged

Clean FID eval for SDXL #86

merged 10 commits into from
Nov 7, 2023

Conversation

jazcollins
Copy link
Contributor

@jazcollins jazcollins commented Oct 16, 2023

This PR makes two changes:

  1. Small fixes necessary to get make FID eval compatible with SDXL
  2. Modify generate() function to allow passing SDXL micro-conditioning as either a tuple (one set of crop/size parameters that will replicated across the batch dimension) or a tensor (batch of different crop/size parameters) - the latter of which is necessary for 1.

@jazcollins jazcollins marked this pull request as ready for review October 16, 2023 21:27
Copy link
Contributor

@Landanjs Landanjs left a comment

Choose a reason for hiding this comment

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

LGTM! Minor comments

diffusion/models/stable_diffusion.py Outdated Show resolved Hide resolved
diffusion/models/stable_diffusion.py Outdated Show resolved Hide resolved
diffusion/models/stable_diffusion.py Outdated Show resolved Hide resolved
diffusion/models/stable_diffusion.py Outdated Show resolved Hide resolved
@jazcollins jazcollins merged commit 01366cc into mosaicml:main Nov 7, 2023
7 checks passed
@jazcollins jazcollins deleted the sdxl-eval branch November 7, 2023 19:33
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.

4 participants