Remove training guidance_scale

1 files changed, 0 insertions, 4 deletions

diff --git a/data/csv.py b/data/csv.py
index d726033..43bf14c 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -108,7 +108,6 @@ def collate_fn(
     dtype: torch.dtype,
     tokenizer: CLIPTokenizer,
     max_token_id_length: Optional[int],
-    with_guidance: bool,
     with_prior_preservation: bool,
     examples,
 ):
@@ -195,7 +194,6 @@ class VlpnDataModule:
         tokenizer: CLIPTokenizer,
         constant_prompt_length: bool = False,
         class_subdir: str = "cls",
-        with_guidance: bool = False,
         num_class_images: int = 1,
         size: int = 768,
         num_buckets: int = 0,
@@ -228,7 +226,6 @@ class VlpnDataModule:
         self.class_root.mkdir(parents=True, exist_ok=True)
         self.placeholder_tokens = placeholder_tokens
         self.num_class_images = num_class_images
-        self.with_guidance = with_guidance
 
         self.constant_prompt_length = constant_prompt_length
         self.max_token_id_length = None
@@ -356,7 +353,6 @@ class VlpnDataModule:
             self.dtype,
             self.tokenizer,
             self.max_token_id_length,
-            self.with_guidance,
             self.num_class_images != 0,
         )