Spaces:

nvidia
/

Cosmos-Predict2

Running on Zero

multimodalart HF Staff commited on 2 days ago

Commit

cbd7c76

verified ·

1 Parent(s): 3e6e28e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,6 @@ import random
 model_14b_id = "nvidia/Cosmos-Predict2-14B-Text2Image"
 pipe_14b = Cosmos2TextToImagePipeline.from_pretrained(model_14b_id,
-                                                      revision="refs/pr/1",
                                                       torch_dtype=torch.bfloat16
                                                      )

 model_14b_id = "nvidia/Cosmos-Predict2-14B-Text2Image"
 pipe_14b = Cosmos2TextToImagePipeline.from_pretrained(model_14b_id,
                                                       torch_dtype=torch.bfloat16
                                                      )