Spaces:
Paused
Paused
Commit
·
e9ed143
1
Parent(s):
566300f
add random seed
Browse files
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=
|
| 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():
|