decrease-inference

#2
by echarlaix HF staff - opened
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