summaryrefslogtreecommitdiffstats
path: root/data
diff options
context:
space:
mode:
Diffstat (limited to 'data')
-rw-r--r--data/csv.py4
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(
108 dtype: torch.dtype, 108 dtype: torch.dtype,
109 tokenizer: CLIPTokenizer, 109 tokenizer: CLIPTokenizer,
110 max_token_id_length: Optional[int], 110 max_token_id_length: Optional[int],
111 with_guidance: bool,
112 with_prior_preservation: bool, 111 with_prior_preservation: bool,
113 examples, 112 examples,
114): 113):
@@ -195,7 +194,6 @@ class VlpnDataModule:
195 tokenizer: CLIPTokenizer, 194 tokenizer: CLIPTokenizer,
196 constant_prompt_length: bool = False, 195 constant_prompt_length: bool = False,
197 class_subdir: str = "cls", 196 class_subdir: str = "cls",
198 with_guidance: bool = False,
199 num_class_images: int = 1, 197 num_class_images: int = 1,
200 size: int = 768, 198 size: int = 768,
201 num_buckets: int = 0, 199 num_buckets: int = 0,
@@ -228,7 +226,6 @@ class VlpnDataModule:
228 self.class_root.mkdir(parents=True, exist_ok=True) 226 self.class_root.mkdir(parents=True, exist_ok=True)
229 self.placeholder_tokens = placeholder_tokens 227 self.placeholder_tokens = placeholder_tokens
230 self.num_class_images = num_class_images 228 self.num_class_images = num_class_images
231 self.with_guidance = with_guidance
232 229
233 self.constant_prompt_length = constant_prompt_length 230 self.constant_prompt_length = constant_prompt_length
234 self.max_token_id_length = None 231 self.max_token_id_length = None
@@ -356,7 +353,6 @@ class VlpnDataModule:
356 self.dtype, 353 self.dtype,
357 self.tokenizer, 354 self.tokenizer,
358 self.max_token_id_length, 355 self.max_token_id_length,
359 self.with_guidance,
360 self.num_class_images != 0, 356 self.num_class_images != 0,
361 ) 357 )
362 358