ds1david commited on
Commit
a09dd26
Β·
1 Parent(s): 0fa6c9c

fixing bugs

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -85,8 +85,7 @@ try:
85
  pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
86
  "stabilityai/stable-diffusion-xl-base-1.0",
87
  torch_dtype=torch_dtype,
88
- use_safetensors=True,
89
- variant="fp16" if device == "cuda" else None
90
  ).to(device)
91
 
92
  # LoRA
 
85
  pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
86
  "stabilityai/stable-diffusion-xl-base-1.0",
87
  torch_dtype=torch_dtype,
88
+ variant="fp32"
 
89
  ).to(device)
90
 
91
  # LoRA