multimodalart HF Staff commited on
Commit
f554964
·
verified ·
1 Parent(s): a4d067b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,10 +7,10 @@ import random
7
  model_14b_id = "diffusers-internal-dev/ct2i-14B"
8
  model_2b_id = "diffusers-internal-dev/ct2i-2B"
9
 
10
- pipe_14b = CosmosTextToImagePipeline.from_pretrained(model_14b_id, torch_dtype=torch.bfloat16)
11
  pipe_14b.to("cuda")
12
 
13
- pipe_2b = CosmosTextToImagePipeline.from_pretrained(model_2b_id, torch_dtype=torch.bfloat16)
14
  pipe_2b.to("cuda")
15
 
16
  scheduler = EDMEulerScheduler(
 
7
  model_14b_id = "diffusers-internal-dev/ct2i-14B"
8
  model_2b_id = "diffusers-internal-dev/ct2i-2B"
9
 
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(