svjack commited on
Commit
1cf9718
·
verified ·
1 Parent(s): 406d477

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- import spaces
3
  from diffusers import StableDiffusionXLPipeline
4
  import numpy as np
5
  import math
@@ -21,7 +21,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
21
  device="cuda"
22
  pipe = pipe.to(device)
23
 
24
- @spaces.GPU
25
  def run(prompt, negative_prompt=None, guidance_scale=7.0, pag_scale=3.0, pag_layers=["mid"], randomize_seed=True, seed=42, lora=None, progress=gr.Progress(track_tqdm=True)):
26
  prompt = prompt.strip()
27
  negative_prompt = negative_prompt.strip() if negative_prompt and negative_prompt.strip() else None
 
1
  import gradio as gr
2
+ #import spaces
3
  from diffusers import StableDiffusionXLPipeline
4
  import numpy as np
5
  import math
 
21
  device="cuda"
22
  pipe = pipe.to(device)
23
 
24
+ #@spaces.GPU
25
  def run(prompt, negative_prompt=None, guidance_scale=7.0, pag_scale=3.0, pag_layers=["mid"], randomize_seed=True, seed=42, lora=None, progress=gr.Progress(track_tqdm=True)):
26
  prompt = prompt.strip()
27
  negative_prompt = negative_prompt.strip() if negative_prompt and negative_prompt.strip() else None