UdacityNoob commited on
Commit
cf11c5a
β€’
1 Parent(s): 994f306

Remove 'steps' from function parameter, use default value 50

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, steps:int):
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]