Spaces:
Paused
Paused
uuu
Browse files
app.py
CHANGED
@@ -34,11 +34,11 @@ ip_adapter_path = './ip-adapter.bin'
|
|
34 |
image_encoder_path = "google/siglip-so400m-patch14-384"
|
35 |
|
36 |
transformer = SD3Transformer2DModel.from_pretrained(
|
37 |
-
model_path, subfolder="transformer", torch_dtype=torch.
|
38 |
)
|
39 |
|
40 |
pipe = StableDiffusion3Pipeline.from_pretrained(
|
41 |
-
model_path, transformer=transformer, torch_dtype=torch.
|
42 |
).to("cuda")
|
43 |
|
44 |
pipe.init_ipadapter(
|
|
|
34 |
image_encoder_path = "google/siglip-so400m-patch14-384"
|
35 |
|
36 |
transformer = SD3Transformer2DModel.from_pretrained(
|
37 |
+
model_path, subfolder="transformer", torch_dtype=torch.float16
|
38 |
)
|
39 |
|
40 |
pipe = StableDiffusion3Pipeline.from_pretrained(
|
41 |
+
model_path, transformer=transformer, torch_dtype=torch.float16
|
42 |
).to("cuda")
|
43 |
|
44 |
pipe.init_ipadapter(
|