Spaces:
Running
on
A100
Running
on
A100
update
Browse files
app.py
CHANGED
@@ -298,6 +298,9 @@ class HunyuanImageApp:
|
|
298 |
|
299 |
self.pipeline.to('cpu')
|
300 |
self.pipeline.refiner_pipeline.to('cuda')
|
|
|
|
|
|
|
301 |
|
302 |
# Use refiner from the main pipeline
|
303 |
refined_image = self.pipeline.refiner_pipeline(
|
@@ -540,7 +543,7 @@ def create_interface(auto_load: bool = True, use_distilled: bool = False, device
|
|
540 |
)
|
541 |
|
542 |
refine_seed = gr.Number(
|
543 |
-
label="Seed", value
|
544 |
info="Random seed for reproducibility"
|
545 |
)
|
546 |
|
|
|
298 |
|
299 |
self.pipeline.to('cpu')
|
300 |
self.pipeline.refiner_pipeline.to('cuda')
|
301 |
+
if seed == -1:
|
302 |
+
import random
|
303 |
+
seed = random.randint(100000, 999999)
|
304 |
|
305 |
# Use refiner from the main pipeline
|
306 |
refined_image = self.pipeline.refiner_pipeline(
|
|
|
543 |
)
|
544 |
|
545 |
refine_seed = gr.Number(
|
546 |
+
label="Seed", value=-1, precision=0,
|
547 |
info="Random seed for reproducibility"
|
548 |
)
|
549 |
|