From 0da1aa90b5d5ca2fcd3359240c24fa458a2ec403 Mon Sep 17 00:00:00 2001 From: Miguel Martin Date: Mon, 5 Jan 2026 17:44:39 -0800 Subject: [PATCH] Fix typo in src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py (#12914) --- src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py b/src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py index 372684e0b5..ea9df999dd 100644 --- a/src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py +++ b/src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py @@ -76,7 +76,7 @@ EXAMPLE_DOC_STRING = """ >>> model_id = "nvidia/Cosmos-Predict2.5-2B" >>> pipe = Cosmos2_5_PredictBasePipeline.from_pretrained( - ... model_id, revision="diffusers/base/pre-trianed", torch_dtype=torch.bfloat16 + ... model_id, revision="diffusers/base/post-trained", torch_dtype=torch.bfloat16 ... ) >>> pipe = pipe.to("cuda")