Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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", "
|
| 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()
|