Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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
|