Spaces:
Running on Zero
Running on Zero
Use Adam inversion by default for concept sliders
Browse files
README.md
CHANGED
|
@@ -35,6 +35,7 @@ and retrieval run on CPU from cached slider masks.
|
|
| 35 |
- `SSR_L0` β subfolder to load, e.g. `L0-20` (variable)
|
| 36 |
- `SSR_CORPUS_REPO` β HF **dataset** repo id with `corpus.npz` + `meta.json` (variable)
|
| 37 |
- `SSR_DEVICE` β `cuda` (variable)
|
|
|
|
| 38 |
- `HF_TOKEN` β a read token (secret) so the Space can read the private corpus dataset
|
| 39 |
|
| 40 |
## Layout
|
|
|
|
| 35 |
- `SSR_L0` β subfolder to load, e.g. `L0-20` (variable)
|
| 36 |
- `SSR_CORPUS_REPO` β HF **dataset** repo id with `corpus.npz` + `meta.json` (variable)
|
| 37 |
- `SSR_DEVICE` β `cuda` (variable)
|
| 38 |
+
- `SSR_INVERSION_METHOD` β `adam` (variable; deployment sets this explicitly)
|
| 39 |
- `HF_TOKEN` β a read token (secret) so the Space can read the private corpus dataset
|
| 40 |
|
| 41 |
## Layout
|
core.py
CHANGED
|
@@ -11,6 +11,7 @@ Configuration (env):
|
|
| 11 |
SSR_CORPUS_REPO HF dataset repo id holding corpus.npz + meta.json (overrides local dir)
|
| 12 |
SSR_CORPUS_DIR local corpus dir (default: demo/corpus)
|
| 13 |
SSR_DEVICE "cuda" / "cpu" (auto-detected)
|
|
|
|
| 14 |
"""
|
| 15 |
from __future__ import annotations
|
| 16 |
|
|
@@ -32,7 +33,7 @@ CORPUS_REPO = os.environ.get("SSR_CORPUS_REPO") # HF dataset repo id, if hosted
|
|
| 32 |
CHECKPOINT = os.environ.get("SSR_CHECKPOINT", os.path.join(HERE, os.pardir, "logs/xps/66a0caf9/checkpoints/last.ckpt"))
|
| 33 |
L0_SUBFOLDER = os.environ.get("SSR_L0") # e.g. "L0-20" when CHECKPOINT is a Hub repo
|
| 34 |
CONFIG = os.environ.get("SSR_CONFIG")
|
| 35 |
-
INVERSION_METHOD = os.environ.get("SSR_INVERSION_METHOD", "
|
| 36 |
DEBUG = os.environ.get("SSR_DEBUG", "1").lower() not in {"0", "false", "no", "off"}
|
| 37 |
|
| 38 |
|
|
|
|
| 11 |
SSR_CORPUS_REPO HF dataset repo id holding corpus.npz + meta.json (overrides local dir)
|
| 12 |
SSR_CORPUS_DIR local corpus dir (default: demo/corpus)
|
| 13 |
SSR_DEVICE "cuda" / "cpu" (auto-detected)
|
| 14 |
+
SSR_INVERSION_METHOD "adam" (default) or "fista"
|
| 15 |
"""
|
| 16 |
from __future__ import annotations
|
| 17 |
|
|
|
|
| 33 |
CHECKPOINT = os.environ.get("SSR_CHECKPOINT", os.path.join(HERE, os.pardir, "logs/xps/66a0caf9/checkpoints/last.ckpt"))
|
| 34 |
L0_SUBFOLDER = os.environ.get("SSR_L0") # e.g. "L0-20" when CHECKPOINT is a Hub repo
|
| 35 |
CONFIG = os.environ.get("SSR_CONFIG")
|
| 36 |
+
INVERSION_METHOD = os.environ.get("SSR_INVERSION_METHOD", "adam")
|
| 37 |
DEBUG = os.environ.get("SSR_DEBUG", "1").lower() not in {"0", "false", "no", "off"}
|
| 38 |
|
| 39 |
|