Hugging Face's logo

Spaces:
Tech-Meld
/
SuperFast_SDXL
Runtime error

App Files Community
Tech-Meld commited on
Commit
d5c20bb
·
verified ·
1 Parent(s): 4dc899e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from diffusers import DiffusionPipeline, LCMScheduler
2
  import gradio as gr
3
 
4
 
5
- pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0").to("cuda")
6
  pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
7
  pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
8
 
 
2
  import gradio as gr
3
 
4
 
5
+ pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0").to("cpu")
6
  pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
7
  pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
8