Hugging Face's logo

Spaces:
harpomaxx
/
deeplili
Sleeping

App Files Community
harpomaxx commited on
Commit
2e6586b
·
1 Parent(s): caf0600

change device from cuda to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ import gradio as gr
13
  guidance_scale = 8.5 # Scale for classifier-free guidance
14
 
15
 
16
- pipe = StableDiffusionPipeline.from_pretrained(PATH,local_files_only=False ).to("cuda")
17
  guidance_scale = 8.5
18
 
19
  def generate_images(prompt, guidance_scale, n_samples, num_inference_steps):
 
13
  guidance_scale = 8.5 # Scale for classifier-free guidance
14
 
15
 
16
+ pipe = StableDiffusionPipeline.from_pretrained(PATH,local_files_only=False ).to("cpu")
17
  guidance_scale = 8.5
18
 
19
  def generate_images(prompt, guidance_scale, n_samples, num_inference_steps):