Rawal Khirodkar commited on
Commit
d053090
·
1 Parent(s): c43d03d

Normal: enable full preload of all 4 sizes; default = 1B (shipped)

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -54,7 +54,7 @@ NORMAL_MODELS = {
54
  "config": os.path.join(CONFIGS_DIR, "sapiens2_5b_normal_metasim_render_people-1024x768.py"),
55
  },
56
  }
57
- DEFAULT_SIZE = "0.4B" # iteration mode — only this is preloaded; others lazy-load on click
58
 
59
  # v1 binary fg/bg TorchScript model.
60
  FG_REPO = "facebook/sapiens-seg-foreground-1b-torchscript"
@@ -96,13 +96,10 @@ def _get_fg_model():
96
  return _fg_model
97
 
98
 
99
- # Iteration mode: only preload the default (0.4B) for fast Space boot.
100
- # Re-enable full preload by uncommenting the loop below.
101
- print("[startup] pre-loading 0.4B (iteration mode) + fg/bg ...")
102
- _get_normal_model(DEFAULT_SIZE)
103
  _get_fg_model()
104
- # for _size in NORMAL_MODELS:
105
- # _get_normal_model(_size)
106
  print("[startup] ready.")
107
 
108
 
 
54
  "config": os.path.join(CONFIGS_DIR, "sapiens2_5b_normal_metasim_render_people-1024x768.py"),
55
  },
56
  }
57
+ DEFAULT_SIZE = "1B"
58
 
59
  # v1 binary fg/bg TorchScript model.
60
  FG_REPO = "facebook/sapiens-seg-foreground-1b-torchscript"
 
96
  return _fg_model
97
 
98
 
99
+ print("[startup] pre-loading all normal sizes + fg/bg ...")
100
+ for _size in NORMAL_MODELS:
101
+ _get_normal_model(_size)
 
102
  _get_fg_model()
 
 
103
  print("[startup] ready.")
104
 
105