KevinNg99 commited on
Commit
ccba969
·
1 Parent(s): e80e082
Files changed (1) hide show
  1. app.py +4 -2
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 = "./ckpts"):
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,