Fix sd feature extractor extensions parser (#62)

yzslab · Sep 21, 2024 · c4033c9 · c4033c9
1 parent 92b0126
commit c4033c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/sd_feature_extraction.py b/utils/sd_feature_extraction.py
@@ -269,7 +269,7 @@ def parse_args():
     parser = argparse.ArgumentParser()
     parser.add_argument("image_dir", type=str)
     parser.add_argument("--output", "-o", type=str, default=None)
-    parser.add_argument("--extensions", "-e", type=str, default=["jpg", "JPG", "jpeg", "JPEG"])
+    parser.add_argument("--extensions", "-e", nargs="+", type=str, default=["jpg", "JPG", "jpeg", "JPEG"])
     parser.add_argument("--image-size", "-s", type=int, default=800)
     parser.add_argument("--seed", type=int, default=42)
     parser.add_argument("--image_list", "--image-list", type=str, default=None)