pngwn HF staff echarlaix HF staff commited on
Commit
270595f
1 Parent(s): dc3112c

decrease-inference (#2)

Browse files

- Decrease number of step (36e8b208fcd766f3cf91f3e0c2eb6a699e7dc1b5)


Co-authored-by: Ella Charlaix <echarlaix@users.noreply.huggingface.co>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def fn(sign, cat):
41
 
42
  response = pipe(starting_text, max_length=(len(starting_text) + random.randint(60, 90)), num_return_sequences=1)
43
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
44
- image = pipe(response[0]["generated_text"], num_inference_steps=25).images[0]
45
  return [image, starting_text]
46
 
47
 
 
41
 
42
  response = pipe(starting_text, max_length=(len(starting_text) + random.randint(60, 90)), num_return_sequences=1)
43
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
44
+ image = pipe(response[0]["generated_text"], num_inference_steps=10).images[0]
45
  return [image, starting_text]
46
 
47