Make image batch size to encode configurable via config.yml

2026-03-03 05:29:12 +00:00 · 2021-09-16 10:51:39 -07:00
parent 41c328dae0
commit 3afe054312
4 changed files with 9 additions and 8 deletions
--- a/src/utils/cli.py
+++ b/src/utils/cli.py
@@ -56,7 +56,8 @@ default_config = {
        },
        'image':
        {
-            'embeddings-file': '.image_embeddings.pt'
+            'embeddings-file': '.image_embeddings.pt',
+            'batch-size': 50
        },
        'music':
        {