Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 =
|
11 |
pipe_14b.to("cuda")
|
12 |
|
13 |
-
pipe_2b =
|
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(
|