Hugging Face's logo

Spaces:
nvidia
/
Cosmos-Predict2
Running on Zero

App Files Community
1
multimodalart HF Staff commited on
Commit
ec066d7
·
verified ·
1 Parent(s): f554964

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ model_2b_id = "diffusers-internal-dev/ct2i-2B"
10
  pipe_14b = Cosmos2TextToImagePipeline.from_pretrained(model_14b_id, torch_dtype=torch.bfloat16)
11
  pipe_14b.to("cuda")
12
 
13
- pipe_2b = Cosmos2TextToImagePipeline.from_pretrained(model_2b_id, torch_dtype=torch.bfloat16)
14
  pipe_2b.to("cuda")
15
 
16
  scheduler = EDMEulerScheduler(
 
10
  pipe_14b = Cosmos2TextToImagePipeline.from_pretrained(model_14b_id, torch_dtype=torch.bfloat16)
11
  pipe_14b.to("cuda")
12
 
13
+ pipe_2b = Cosmos2TextToImagePipeline.from_pretrained(model_2b_id, text_encoder=pipe_14b.text_encoder, safety_checker=pipe_14b.safety_checker, vae=pipe_14b.vae, tokenizer=pipe_14b.tokenizer, torch_dtype=torch.bfloat16)
14
  pipe_2b.to("cuda")
15
 
16
  scheduler = EDMEulerScheduler(