hotfix(missing-var): h_flip (#86)

cloneofsimo · Dec 27, 2022 · 3400a2b · 3400a2b
1 parent 66c18d3
commit 3400a2b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train_lora_dreambooth.py b/train_lora_dreambooth.py
@@ -63,6 +63,7 @@ def __init__(
         size=512,
         center_crop=False,
         color_jitter=False,
+        h_flip=False,
         resize=False,
     ):
         self.size = size