Fix display name for sharded model file

0cc4m · Sep 9, 2023 · 85b009c · 85b009c
1 parent 766e582
commit 85b009c
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/webui/session.py b/webui/session.py
@@ -310,7 +310,16 @@ def api_populate(self):
 
         # Add model info
 
-        model_str = os.path.splitext(os.path.basename(model.config.model_path))[0] + "\n"
+        def _common_chars(names):
+            cname = max(names, key=len)
+            for x in names:
+                for p, c in enumerate(x):
+                    if c != cname[p] and cname[p] != "*": cname = cname[:p] + "*" + cname[p + 1:]
+            return cname
+
+        mp = model.config.model_path if isinstance(model.config.model_path, str) else _common_chars(model.config.model_path)
+
+        model_str = os.path.splitext(os.path.basename(mp))[0] + "\n"
         model_str += f"Sequence length: {model.config.max_seq_len}\n"
 
         dic["model_info"] = model_str.strip()