From 3e961a9860af6226e7035b7406e386976fa1e7f8 Mon Sep 17 00:00:00 2001 From: Will Miao Date: Fri, 12 Jun 2026 08:57:25 +0800 Subject: [PATCH] fix(stats): load embeddings from saved stats on startup _load_stats() was missing the embeddings section, so on every restart the embeddings usage tracking hash would start from an empty dict. This caused all previously saved embedding usage data to appear reset. Added the missing load path for the 'embeddings' key, parallel to the existing checkpoints and loras loading logic. --- py/utils/usage_stats.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/py/utils/usage_stats.py b/py/utils/usage_stats.py index 066b8314..774b887e 100644 --- a/py/utils/usage_stats.py +++ b/py/utils/usage_stats.py @@ -221,6 +221,9 @@ class UsageStats: if "loras" in loaded_stats and isinstance(loaded_stats["loras"], dict): self.stats["loras"] = loaded_stats["loras"] + if "embeddings" in loaded_stats and isinstance(loaded_stats["embeddings"], dict): + self.stats["embeddings"] = loaded_stats["embeddings"] + if "total_executions" in loaded_stats: self.stats["total_executions"] = loaded_stats["total_executions"]