add repo url to models info

smallcloudai · May 16, 2024 · 9fb5939 · 9fb5939
1 parent ce839e6
commit 9fb5939
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/refact_webgui/webgui/selfhost_model_assigner.py b/refact_webgui/webgui/selfhost_model_assigner.py
@@ -247,6 +247,7 @@ def models_info(self):
                 "available_n_ctx": available_n_ctx,
                 "is_deprecated": bool(rec.get("deprecated", False)),
                 "repo_status": self._models_repo_status[k],
+                "repo_url": f"https://huggingface.co/{rec['model_path']}",
             })
         return {"models": info}