Spaces:

nvidia
/

Cosmos-Predict2

Running on Zero

multimodalart HF Staff commited on 7 days ago

Commit

1c639df

verified ·

1 Parent(s): e57b5c6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import torch
 from diffusers import Cosmos2TextToImagePipeline, EDMEulerScheduler
 import random
-model_14b_id = "diffusers-internal-dev/ct2i-14B"
-model_2b_id = "diffusers-internal-dev/ct2i-2B"
 pipe_14b = Cosmos2TextToImagePipeline.from_pretrained(model_14b_id,
                                                       revision="refs/pr/1",

 from diffusers import Cosmos2TextToImagePipeline, EDMEulerScheduler
 import random
+model_14b_id = "nvidia/Cosmos-Predict2-14B-Text2Image"
+model_2b_id = "nvidia/Cosmos-Predict2-2B-Text2Image"
 pipe_14b = Cosmos2TextToImagePipeline.from_pretrained(model_14b_id,
                                                       revision="refs/pr/1",