multimodalart HF Staff commited on
Commit
0071360
·
verified ·
1 Parent(s): abfc3df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -12,16 +12,16 @@ pipe_14b = Cosmos2TextToImagePipeline.from_pretrained(model_14b_id,
12
  torch_dtype=torch.bfloat16
13
  )
14
 
15
- #pipe_2b = Cosmos2TextToImagePipeline.from_pretrained(model_2b_id,
16
- # revision="refs/pr/2",
17
- # text_encoder=pipe_14b.text_encoder,
18
- # safety_checker=pipe_14b.safety_checker,
19
- # vae=pipe_14b.vae,
20
- # tokenizer=pipe_14b.tokenizer,
21
- # torch_dtype=torch.bfloat16
22
- # )
23
  pipe_14b.to("cuda")
24
- #pipe_2b.to("cuda")
25
 
26
  scheduler = EDMEulerScheduler(
27
  sigma_min=0.002,
 
12
  torch_dtype=torch.bfloat16
13
  )
14
 
15
+ pipe_2b = Cosmos2TextToImagePipeline.from_pretrained(model_2b_id,
16
+ revision="refs/pr/2",
17
+ text_encoder=pipe_14b.text_encoder,
18
+ safety_checker=pipe_14b.safety_checker,
19
+ vae=pipe_14b.vae,
20
+ tokenizer=pipe_14b.tokenizer,
21
+ torch_dtype=torch.bfloat16
22
+ )
23
  pipe_14b.to("cuda")
24
+ pipe_2b.to("cuda")
25
 
26
  scheduler = EDMEulerScheduler(
27
  sigma_min=0.002,