Nick088 commited on
Commit
87f4d26
·
verified ·
1 Parent(s): 1dee8d8

fix example error by cache

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -20,6 +20,8 @@ else:
20
 
21
  MAX_SEED = np.iinfo(np.int32).max
22
 
 
 
23
 
24
  # Initialize the pipeline and download the sd3 medium model
25
  pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
@@ -126,9 +128,10 @@ with gr.Blocks(css=css) as demo:
126
 
127
  gr.Examples(
128
  examples=examples,
129
- inputs=[prompt, enhance_prompt],
130
- outputs=[result, better_prompt],
131
  fn=generate_image,
 
 
 
132
  )
133
 
134
  gr.on(
 
20
 
21
  MAX_SEED = np.iinfo(np.int32).max
22
 
23
+ CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0") == "1"
24
+
25
 
26
  # Initialize the pipeline and download the sd3 medium model
27
  pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
 
128
 
129
  gr.Examples(
130
  examples=examples,
 
 
131
  fn=generate_image,
132
+ inputs=[prompt, enhance_prompt, negative_prompt, num_inference_steps, guidance_scale, height, width, seed, num_images_per_prompt],
133
+ outputs=[result, better_prompt],
134
+ cache_examples=CACHE_EXAMPLES
135
  )
136
 
137
  gr.on(