diff --git a/scripts/demo/sampling.py b/scripts/demo/sampling.py index 4dca18d..9bf0dff 100644 --- a/scripts/demo/sampling.py +++ b/scripts/demo/sampling.py @@ -237,8 +237,12 @@ if __name__ == "__main__": ) seed_everything(seed) + lowvram_mode = st.checkbox("Low vram mode", True) + save_locally, save_path = init_save_locally(os.path.join(SAVE_PATH, str(version))) - state = init_st(model_specs[version_enum], load_filter=True) + state = init_st( + model_specs[version_enum], load_filter=True, lowvram_mode=lowvram_mode + ) model = state["model"] is_legacy = specs.is_legacy