Add hack to fix error when metadata.txt has tab indentation

LagoLunatic · Sep 13, 2024 · 5d114e4 · 5d114e4
1 parent 7f3f12b
commit 5d114e4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/customizer.py b/customizer.py
@@ -69,6 +69,9 @@ def get_model_metadata(custom_model_name):
       if old_format_match and not new_format_match:
         metadata_str = re.sub(r"^( +\S[^:]*: +)\"?([0-9A-F]{6})\"?$", r"\g<1>0x\g<2>", metadata_str, flags = re.IGNORECASE | re.MULTILINE)
 
+      # Hack to remove tab indentation because it's not valid YAML.
+      metadata_str = re.sub(r"\t", r"  ", metadata_str, flags = re.IGNORECASE | re.MULTILINE)
+
       metadata = yaml.load(metadata_str)
     except Exception as e:
       error_message = str(e)