Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -10,11 +10,11 @@ import cv2
|
|
10 |
# Load the SDXL pipelines
|
11 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
12 |
pipe_tshirt = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
13 |
-
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.
|
14 |
).to(device)
|
15 |
|
16 |
pipe_design = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
17 |
-
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.
|
18 |
).to(device)
|
19 |
|
20 |
# Generate T-shirt image
|
|
|
10 |
# Load the SDXL pipelines
|
11 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
12 |
pipe_tshirt = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
13 |
+
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float32
|
14 |
).to(device)
|
15 |
|
16 |
pipe_design = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
17 |
+
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float32
|
18 |
).to(device)
|
19 |
|
20 |
# Generate T-shirt image
|