Spaces:
Runtime error
Runtime error
UdacityNoob
commited on
Commit
β’
cf11c5a
1
Parent(s):
994f306
Remove 'steps' from function parameter, use default value 50
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4",
|
|
12 |
# pipe = pipe.to('cuda')
|
13 |
|
14 |
# define gradio function
|
15 |
-
def generate(prompt:str, seed:int, guidance:float
|
16 |
generator = torch.Generator("cuda").manual_seed(int(seed))
|
17 |
with autocast("cuda"):
|
18 |
image = pipe(prompt=prompt, generator=generator, guidance_scale=guidance, steps=50).images[0]
|
|
|
12 |
# pipe = pipe.to('cuda')
|
13 |
|
14 |
# define gradio function
|
15 |
+
def generate(prompt:str, seed:int, guidance:float):
|
16 |
generator = torch.Generator("cuda").manual_seed(int(seed))
|
17 |
with autocast("cuda"):
|
18 |
image = pipe(prompt=prompt, generator=generator, guidance_scale=guidance, steps=50).images[0]
|