Skip to content

Commit 0da1aa9

Browse files
Fix typo in src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py (#12914)
1 parent 5ffb658 commit 0da1aa9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def retrieve_latents(
7676
7777
>>> model_id = "nvidia/Cosmos-Predict2.5-2B"
7878
>>> pipe = Cosmos2_5_PredictBasePipeline.from_pretrained(
79-
... model_id, revision="diffusers/base/pre-trianed", torch_dtype=torch.bfloat16
79+
... model_id, revision="diffusers/base/post-trained", torch_dtype=torch.bfloat16
8080
... )
8181
>>> pipe = pipe.to("cuda")
8282

0 commit comments

Comments
 (0)