Cosmos Predict2.5 Base: inference pipeline, scheduler & chkpt conversion #12852
miguelmartin75
changed the title Cosmos/predict2.5 base pr ready Cosmos Predict2.5 Base Model 56 days ago
miguelmartin75
changed the title Cosmos Predict2.5 Base Model Cosmos Predict2.5 Base: inference pipeline, checkpoint conversion & scheduler 56 days ago
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
cosmos predict2.5 base: convert chkpt & pipeline
83d2b81c
scheduler cleanup
4395869d
simplify inference pipeline
e6e278e6
cleanup scheduler + tests
dd6f5406
Basic tests for flow unipc
828788ec
working b2b inference
899be86b
Rename everything
2cc2b564
Tests for pipeline present, but not working (predict2 also not working)
04f23e88
docstring update
824fffa1
wrapper pipelines + make style
bae477a1
remove unnecessary files
232a8168
UniPCMultistep: support use_karras_sigmas=True and use_flow_sigmas=True
1a132f2b
use UniPCMultistepScheduler + fix tests for pipeline
98082204
Remove FlowUniPCMultistepScheduler
abba01c9
UniPCMultistepScheduler for use_flow_sigmas=True & use_karras_sigmas=…
b9a35f5f
num_inference_steps=36 due to bug in scheduler used by predict2.5
dd429ef3
Address comments
b76f9f20
make style + make fix-copies
735fb0e9
fix tests + remove references to old pipelines
46f79166
yiyixuxu
approved these changes
on 2025-12-18
address comments
5c28b087
add revision in from_pretrained call
80817973
fix tests
d1dab59d
sayakpaul
merged
b5309683
into main 54 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub