diffusers
Cosmos Predict2.5 Base: inference pipeline, scheduler & chkpt conversion
#12852
Merged

Cosmos Predict2.5 Base: inference pipeline, scheduler & chkpt conversion #12852

miguelmartin75
miguelmartin75 miguelmartin75 changed the title Cosmos/predict2.5 base pr ready Cosmos Predict2.5 Base Model 56 days ago
miguelmartin75 miguelmartin75 changed the title Cosmos Predict2.5 Base Model Cosmos Predict2.5 Base: inference pipeline, checkpoint conversion & scheduler 56 days ago
miguelmartin75 miguelmartin75 changed the title Cosmos Predict2.5 Base: inference pipeline, checkpoint conversion & scheduler Cosmos Predict2.5 Base: inference pipeline, scheduler & chkpt conversion 56 days ago
miguelmartin75
miguelmartin75 commented on 2025-12-17
sayakpaul
sayakpaul commented on 2025-12-17
sayakpaul sayakpaul requested a review from DN6 DN6 56 days ago
sayakpaul sayakpaul requested a review from yiyixuxu yiyixuxu 56 days ago
HuggingFaceDocBuilderDev
yiyixuxu
yiyixuxu commented on 2025-12-17
miguelmartin75 miguelmartin75 force pushed from d969b9a5 to 94d2b41e 55 days ago
miguelmartin75 miguelmartin75 force pushed from 4133c68b to 48f373b7 55 days ago
miguelmartin75 cosmos predict2.5 base: convert chkpt & pipeline
83d2b81c
miguelmartin75 scheduler cleanup
4395869d
miguelmartin75 simplify inference pipeline
e6e278e6
miguelmartin75 cleanup scheduler + tests
dd6f5406
miguelmartin75 Basic tests for flow unipc
828788ec
miguelmartin75 working b2b inference
899be86b
miguelmartin75 Rename everything
2cc2b564
miguelmartin75 Tests for pipeline present, but not working (predict2 also not working)
04f23e88
miguelmartin75 docstring update
824fffa1
miguelmartin75 wrapper pipelines + make style
bae477a1
miguelmartin75 remove unnecessary files
232a8168
miguelmartin75 UniPCMultistep: support use_karras_sigmas=True and use_flow_sigmas=True
1a132f2b
miguelmartin75 use UniPCMultistepScheduler + fix tests for pipeline
98082204
miguelmartin75 Remove FlowUniPCMultistepScheduler
abba01c9
miguelmartin75 UniPCMultistepScheduler for use_flow_sigmas=True & use_karras_sigmas=…
b9a35f5f
miguelmartin75 num_inference_steps=36 due to bug in scheduler used by predict2.5
dd429ef3
miguelmartin75 miguelmartin75 force pushed from 48f373b7 to c14a3dad 55 days ago
miguelmartin75 Address comments
b76f9f20
miguelmartin75 miguelmartin75 force pushed from c14a3dad to b76f9f20 55 days ago
miguelmartin75
sayakpaul
github-actions
sayakpaul
miguelmartin75 make style + make fix-copies
735fb0e9
miguelmartin75 miguelmartin75 force pushed from 5f41bc1d to 735fb0e9 55 days ago
yiyixuxu
yiyixuxu commented on 2025-12-18
yiyixuxu
yiyixuxu commented on 2025-12-18
miguelmartin75 fix tests + remove references to old pipelines
46f79166
yiyixuxu
yiyixuxu commented on 2025-12-18
yiyixuxu
yiyixuxu commented on 2025-12-18
yiyixuxu
yiyixuxu commented on 2025-12-18
yiyixuxu
yiyixuxu commented on 2025-12-18
yiyixuxu
yiyixuxu approved these changes on 2025-12-18
miguelmartin75 address comments
5c28b087
miguelmartin75 add revision in from_pretrained call
80817973
miguelmartin75 fix tests
d1dab59d
sayakpaul sayakpaul merged b5309683 into main 54 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone