listen2you003 commited on
Commit
77883c8
·
1 Parent(s): d44ac9b

change randomseed

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -403,6 +403,8 @@ def inference(prompt, ref_images, seed, size_level, infer_func=None):
403
  if seed == -1:
404
  import random
405
  random_seed = random.randint(0, 2**32 - 1)
 
 
406
 
407
  image = infer_func(
408
  prompt,
 
403
  if seed == -1:
404
  import random
405
  random_seed = random.randint(0, 2**32 - 1)
406
+ else:
407
+ random_seed = seed
408
 
409
  image = infer_func(
410
  prompt,
requirements.txt CHANGED
@@ -9,4 +9,5 @@ diffusers
9
  peft
10
  opencv-python
11
  sentencepiece
12
- boto3
 
 
9
  peft
10
  opencv-python
11
  sentencepiece
12
+ boto3
13
+ torchvision