thecollabagepatch commited on
Commit
c211cf7
·
1 Parent(s): f0823f1

cleaning up /model/select when switching to base model

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -581,9 +581,15 @@ def model_select(req: ModelSelect):
581
  if req.dry_run:
582
  return {"ok": True, "dry_run": True, **validation_result}
583
 
584
- if req.ckpt_step == "none": # user asked for stock base
585
- asset_manager.clear_assets() # implement .clear_assets() to set embeds/centroids to None
586
- _sync_assets_globals_from_manager()
 
 
 
 
 
 
587
 
588
  # 2) Handle jam policy
589
  if _any_jam_running():
 
581
  if req.dry_run:
582
  return {"ok": True, "dry_run": True, **validation_result}
583
 
584
+ if isinstance(req.step, str) and req.step.lower() == "none": # user asked for stock base
585
+ # Clear any resident finetune assets so /model/config reflects "no assets"
586
+ asset_manager.mean_embed = None
587
+ asset_manager.centroids = None
588
+ asset_manager.assets_repo_id = None
589
+ # keep module-level mirrors in sync
590
+ _MEAN_EMBED = None
591
+ _CENTROIDS = None
592
+ _ASSETS_REPO_ID = None
593
 
594
  # 2) Handle jam policy
595
  if _any_jam_running():