Miguel Martin a1816166a5 Cosmos Transfer2.5 inference pipeline: general/{seg, depth, blur, edge} (#13066) ...

* initial conversion script

* cosmos control net block

* CosmosAttention

* base model conversion

* wip

* pipeline updates

* convert controlnet

* pipeline: working without controls

* wip

* debugging

* Almost working

* temp

* control working

* cleanup + detail on neg_encoder_hidden_states

* convert edge

* pos emb for control latents

* convert all chkpts

* resolve TODOs

* remove prints

* Docs

* add siglip image reference encoder

* Add unit tests

* controlnet: add duplicate layers

* Additional tests

* skip less

* skip less

* remove image_ref

* minor

* docs

* remove skipped test in transfer

* Don't crash process

* formatting

* revert some changes

* remove skipped test

* make style

* Address comment + fix example

* CosmosAttnProcessor2_0 revert + CosmosAttnProcessor2_5 changes

* make style

* make fix-copies

2026-02-11 18:33:09 -10:00

..

__init__.py

Cosmos Transfer2.5 inference pipeline: general/{seg, depth, blur, edge} (#13066)

2026-02-11 18:33:09 -10:00

test_models_controlnet_cosmos.py

Cosmos Transfer2.5 inference pipeline: general/{seg, depth, blur, edge} (#13066)

2026-02-11 18:33:09 -10:00