fffiloni commited on
Commit
af3ec86
·
verified ·
1 Parent(s): 06a2a33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,7 +51,7 @@ DEFAULT_SHIFT = float(os.getenv("SCAIL_SAMPLE_SHIFT", "3.0"))
51
  DEFAULT_GUIDE_SCALE = float(os.getenv("SCAIL_GUIDE_SCALE", "5.0"))
52
  DEFAULT_SOLVER = os.getenv("SCAIL_SAMPLE_SOLVER", "unipc")
53
  AUTO_CONVERT = os.getenv("SCAIL_AUTO_CONVERT", "1") == "1"
54
- PRELOAD_PIPELINE = os.getenv("SCAIL_PRELOAD_PIPELINE", "0") == "1"
55
  STAGE_SAFETENSORS_FOR_LOAD = os.getenv("SCAIL_STAGE_SAFETENSORS_FOR_LOAD", "1") == "1"
56
  CLIP_CKPT_NAME = "models_clip_open-clip-xlm-roberta-large-vit-huge-14-onlyvisual.pth"
57
  ORIGINAL_DIT_REL_PATH = "model/1/fsdp2_rank_0000_checkpoint.pt"
@@ -911,4 +911,4 @@ if __name__ == "__main__":
911
  _prepare_runtime_for_startup()
912
  if PRELOAD_PIPELINE:
913
  _prepare_pipeline_for_startup()
914
- build_ui().queue(max_size=8).launch()
 
51
  DEFAULT_GUIDE_SCALE = float(os.getenv("SCAIL_GUIDE_SCALE", "5.0"))
52
  DEFAULT_SOLVER = os.getenv("SCAIL_SAMPLE_SOLVER", "unipc")
53
  AUTO_CONVERT = os.getenv("SCAIL_AUTO_CONVERT", "1") == "1"
54
+ PRELOAD_PIPELINE = os.getenv("SCAIL_PRELOAD_PIPELINE", "1") == "1"
55
  STAGE_SAFETENSORS_FOR_LOAD = os.getenv("SCAIL_STAGE_SAFETENSORS_FOR_LOAD", "1") == "1"
56
  CLIP_CKPT_NAME = "models_clip_open-clip-xlm-roberta-large-vit-huge-14-onlyvisual.pth"
57
  ORIGINAL_DIT_REL_PATH = "model/1/fsdp2_rank_0000_checkpoint.pt"
 
911
  _prepare_runtime_for_startup()
912
  if PRELOAD_PIPELINE:
913
  _prepare_pipeline_for_startup()
914
+ build_ui().queue(max_size=8).launch()