multimodalart HF Staff commited on
Commit
1343ee5
·
verified ·
1 Parent(s): 510eeee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ theme = gr.themes.Base(
15
  pipe = StableDiffusionXLPipeline.from_pretrained(
16
  "corcelio/mobius",
17
  custom_pipeline="multimodalart/sdxl_perturbed_attention_guidance",
18
- torch_dtype=torch.float16
 
19
  )
20
 
21
  device="cuda"
 
15
  pipe = StableDiffusionXLPipeline.from_pretrained(
16
  "corcelio/mobius",
17
  custom_pipeline="multimodalart/sdxl_perturbed_attention_guidance",
18
+ torch_dtype=torch.float16,
19
+ use_safetensors=True
20
  )
21
 
22
  device="cuda"