Spaces:
Running
on
A100
Running
on
A100
update
Browse files
app.py
CHANGED
@@ -48,10 +48,12 @@ except ImportError as e:
|
|
48 |
print("For checkpoint downloads, also install: pip install -U 'huggingface_hub[cli]' modelscope")
|
49 |
sys.exit(1)
|
50 |
|
|
|
|
|
51 |
class CheckpointDownloader:
|
52 |
"""Handles downloading of all required checkpoints for HunyuanImage."""
|
53 |
|
54 |
-
def __init__(self, base_dir: str =
|
55 |
self.base_dir = Path(base_dir)
|
56 |
self.base_dir.mkdir(exist_ok=True)
|
57 |
|
@@ -418,7 +420,7 @@ def create_interface(auto_load: bool = True, use_distilled: bool = False, device
|
|
418 |
with gr.Row():
|
419 |
seed = gr.Number(
|
420 |
label="Seed", value=649151, precision=0,
|
421 |
-
info="Random seed for reproducibility"
|
422 |
)
|
423 |
use_reprompt = gr.Checkbox(
|
424 |
label="Use Reprompt", value=True,
|
|
|
48 |
print("For checkpoint downloads, also install: pip install -U 'huggingface_hub[cli]' modelscope")
|
49 |
sys.exit(1)
|
50 |
|
51 |
+
|
52 |
+
BASE_DIR = os.environ.get('HUNYUANIMAGE_V2_1_MODEL_ROOT', './ckpts')
|
53 |
class CheckpointDownloader:
|
54 |
"""Handles downloading of all required checkpoints for HunyuanImage."""
|
55 |
|
56 |
+
def __init__(self, base_dir: str = BASE_DIR):
|
57 |
self.base_dir = Path(base_dir)
|
58 |
self.base_dir.mkdir(exist_ok=True)
|
59 |
|
|
|
420 |
with gr.Row():
|
421 |
seed = gr.Number(
|
422 |
label="Seed", value=649151, precision=0,
|
423 |
+
info="Random seed for reproducibility. (-1 for random seed)"
|
424 |
)
|
425 |
use_reprompt = gr.Checkbox(
|
426 |
label="Use Reprompt", value=True,
|