listen2you003 commited on
Commit
e9ed143
·
1 Parent(s): 566300f

add random seed

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -400,6 +400,10 @@ def prepare_infer_func():
400
  def inference(prompt, ref_images, seed, size_level, infer_func=None):
401
  start_time = time.time()
402
 
 
 
 
 
403
  image = infer_func(
404
  prompt,
405
  negative_prompt="",
@@ -407,13 +411,13 @@ def inference(prompt, ref_images, seed, size_level, infer_func=None):
407
  num_samples=1,
408
  num_steps=28,
409
  cfg_guidance=6.0,
410
- seed=seed,
411
  show_progress=True,
412
  size_level=size_level,
413
  )[0]
414
 
415
  print(f"Time taken: {time.time() - start_time:.2f} seconds")
416
- return image
417
 
418
 
419
  def create_demo():
 
400
  def inference(prompt, ref_images, seed, size_level, infer_func=None):
401
  start_time = time.time()
402
 
403
+ if seed == -1:
404
+ import random
405
+ random_seed = random.randint(0, 2**32 - 1)
406
+
407
  image = infer_func(
408
  prompt,
409
  negative_prompt="",
 
411
  num_samples=1,
412
  num_steps=28,
413
  cfg_guidance=6.0,
414
+ seed=random_seed,
415
  show_progress=True,
416
  size_level=size_level,
417
  )[0]
418
 
419
  print(f"Time taken: {time.time() - start_time:.2f} seconds")
420
+ return image, random_seed
421
 
422
 
423
  def create_demo():