Skip to content

Cosmos Predict2 #11695

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.

Sign up for GitHub

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

Jump to bottom
Open
wants to merge 21 commits into
base: main
Choose a base branch
from
Open

Cosmos Predict2 #11695

wants to merge 21 commits into from
+2,471 −60

Conversation

a-r-r-o-w
Copy link
Member

@a-r-r-o-w a-r-r-o-w commented Jun 11, 2025 •
edited
Loading

The cosmos is within us. We are made of star-stuff. We are a way for the universe to know itself.

cc @pjannaty @chenhsuanlin @fitsumreda @asfiyab-nvidia @amolfasale

@a-r-r-o-w a-r-r-o-w requested a review from yiyixuxu June 11, 2025 14:29
@a-r-r-o-w
Copy link
Member Author

a-r-r-o-w commented Jun 11, 2025 •
edited
Loading

@yiyixuxu This PR contains the version that works with FlowMatchEulerDiscreteScheduler. For the version that works with EDMEulerDiscrete, it is in this branch.

We will probably be sticking with this PR IIUC from our discussion, so I'll update the scheduler configs once you confirm (currently only the text2image pipeline has been updated, so will update video2world soon too)

PRs:

@HuggingFaceDocBuilderDev
Copy link

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

pjannaty
pjannaty approved these changes Jun 11, 2025
View reviewed changes
Copy link

@pjannaty pjannaty left a comment

Choose a reason for hiding this comment

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

LGTM! Thank you, Aryan and the HuggingFace team!

yiyixuxu
yiyixuxu approved these changes Jun 11, 2025
View reviewed changes
Copy link
Collaborator

@yiyixuxu yiyixuxu left a comment

Choose a reason for hiding this comment

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

looks good!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@yiyixuxu yiyixuxu yiyixuxu approved these changes

@pjannaty pjannaty pjannaty approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
4 participants
@a-r-r-o-w @HuggingFaceDocBuilderDev @yiyixuxu @pjannaty