fix metadata.min

ControlNet · Apr 7, 2024 · 679cc60 · 679cc60
1 parent c864d1e
commit 679cc60
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -25,7 +25,7 @@
     args = parser.parse_args()
     config = toml.load(args.config)
 
-    if os.path.exists(os.path.join(args.data_root, "metadata.min.json")):
+    if not os.path.exists(os.path.join(args.data_root, "metadata.min.json")):
         generate_metadata_min(args.data_root)
 
     learning_rate = config["optimizer"]["learning_rate"]

diff --git a/utils.py b/utils.py
@@ -228,6 +228,7 @@ def generate_metadata_min(data_root: str):
     metadata_min = []
     for meta in metadata_full:
         del meta["timestamps"]
+        del meta["transcript"]
         metadata_min.append(meta)
     with open(os.path.join(data_root, "metadata.min.json"), "w") as f:
         json.dump(metadata_min, f)