Merge pull request #112 from space-nuko/normalize-model-filenames

Escape commas in model filenames
kohya-ss · Feb 16, 2023 · 822f213 · 822f213
2 parents d0a72e1 + f098335
commit 822f213
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scripts/model_util.py b/scripts/model_util.py
@@ -246,6 +246,10 @@ def get_all_models(paths, sort_by, filter_by):
 
     name = os.path.splitext(os.path.basename(filename))[0]
 
+    # Commas in the model name will mess up infotext restoration since the
+    # infotext is delimited by commas
+    name = name.replace(",", "_")
+
     # Prevent a hypothetical "None.pt" from being listed.
     if name != "None":
       full_name = name + f"({model_hash[0:12]})"