multimodalart HF Staff commited on
Commit
3a1268c
·
verified ·
1 Parent(s): 0529abc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ pipe_14b = Cosmos2TextToImagePipeline.from_pretrained(model_14b_id,
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
 
 
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