gaur3009 commited on
Commit
5a421d5
·
verified ·
1 Parent(s): d4f8ad4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.float16
14
  ).to(device)
15
 
16
  pipe_design = StableDiffusionXLImg2ImgPipeline.from_pretrained(
17
- "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
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