pcuenq HF staff commited on
Commit
d4aa90a
1 Parent(s): 3cc2038
Files changed (1) hide show
  1. app.py +14 -3
app.py CHANGED
@@ -198,7 +198,8 @@ def sample(model, model_inputs, latent_shape,
198
 
199
  # -----
200
 
201
- def infer(prompt, negative_prompt):
 
202
  text = tokenizer.tokenize([prompt] * latent_shape[0]).to(device)
203
  with torch.inference_mode():
204
  if negative_prompt:
@@ -425,8 +426,18 @@ with block:
425
  label="Generated images", show_label=False, elem_id="gallery"
426
  ).style(grid=[2], height="auto")
427
 
428
- text.submit(infer, inputs=[text, negative], outputs=gallery)
429
- btn.click(infer, inputs=[text, negative], outputs=gallery)
 
 
 
 
 
 
 
 
 
 
430
 
431
  gr.HTML(
432
  """
 
198
 
199
  # -----
200
 
201
+ def infer(prompt, negative_prompt, seed):
202
+ torch.manual_seed(seed)
203
  text = tokenizer.tokenize([prompt] * latent_shape[0]).to(device)
204
  with torch.inference_mode():
205
  if negative_prompt:
 
426
  label="Generated images", show_label=False, elem_id="gallery"
427
  ).style(grid=[2], height="auto")
428
 
429
+ with gr.Group():
430
+ with gr.Accordion("Advanced settings", open=False):
431
+ seed = gr.Slider(
432
+ label="Seed",
433
+ minimum=0,
434
+ maximum=2147483647,
435
+ step=1,
436
+ randomize=True,
437
+ )
438
+
439
+ text.submit(infer, inputs=[text, negative, seed], outputs=gallery)
440
+ btn.click(infer, inputs=[text, negative, seed], outputs=gallery)
441
 
442
  gr.HTML(
443
  """