PR #13924 multi-GPU VAE Fix for Cosmos 3

multi-GPU VAE Fix for Cosmos 3 #13924

yiyixuxu merged 2 commits into huggingface:main from atharvajoshi10:multi-gpu-support

fix(cosmos3): pin VAE latent norm buffers to encode output device

9f099d42

github-actions added documentation

github-actions added models

github-actions added pipelines

github-actions added size/L

JingyaHuang requested a review from

yiyixuxu 17 days ago

JingyaHuang requested a review from

sayakpaul 17 days ago

atharvajoshi10 force pushed from 33ea463d to 9f099d42 17 days ago

github-actions removed documentation

github-actions removed models

github-actions removed size/L

github-actions added size/S

atharvajoshi10 changed the title ~~Context parallelism + multi-GPU VAE Fix for Cosmos 3~~ multi-GPU VAE Fix for Cosmos 3 17 days ago

sayakpaul approved these changes on 2026-06-12

Merge branch 'main' into multi-gpu-support

219260c7

yiyixuxu merged bd2c9195 into main 4 days ago

Reviewers

sayakpaul

yiyixuxu

Assignees

No one assigned

Labels

size/S pipelines

Milestone

No milestone