Pliploop commited on
Commit
fbdbff2
Β·
verified Β·
1 Parent(s): 080e646

Use Adam inversion by default for concept sliders

Browse files
Files changed (2) hide show
  1. README.md +1 -0
  2. core.py +2 -1
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", "fista")
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