update cache_latents/text_encoder_outputs

2026-04-08 22:35:09 +00:00 · 2024-10-13 16:27:22 +09:00
parent 5bb9f7fb1a
commit 74228c9953
3 changed files with 165 additions and 160 deletions
--- a/library/strategy_base.py
+++ b/library/strategy_base.py
@@ -325,7 +325,7 @@ class TextEncoderOutputsCachingStrategy:
    def __init__(
        self,
        cache_to_disk: bool,
-        batch_size: int,
+        batch_size: Optional[int],
        skip_disk_cache_validity_check: bool,
        is_partial: bool = False,
        is_weighted: bool = False,