Skip to content

Refactor/private generate method#958

Closed
AnnaKwa wants to merge 46 commits intomainfrom
refactor/private_generate_method
Closed

Refactor/private generate method#958
AnnaKwa wants to merge 46 commits intomainfrom
refactor/private_generate_method

Conversation

@AnnaKwa
Copy link
Contributor

@AnnaKwa AnnaKwa commented Mar 11, 2026

I made some changes to address this comment since they were fresh in my mind: #954 (comment)

This PR refactors CascadePredictor.generate to wrap previous step's output into a BatchData object so it can call the underlying model's generate_on_batch_no_target instead of generate.

This allows us to make the DiffusionModel.generate method private.

@AnnaKwa AnnaKwa closed this Mar 11, 2026
@AnnaKwa AnnaKwa deleted the refactor/private_generate_method branch March 11, 2026 21:52
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.

1 participant