summaryrefslogtreecommitdiffstats
path: root/data
diff options
context:
space:
mode:
Diffstat (limited to 'data')
-rw-r--r--data/csv.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/data/csv.py b/data/csv.py
index 55a1988..d9f9db8 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -118,7 +118,7 @@ class VlpnDataModule():
118 size: int = 768, 118 size: int = 768,
119 num_buckets: int = 0, 119 num_buckets: int = 0,
120 bucket_step_size: int = 64, 120 bucket_step_size: int = 64,
121 max_pixels_per_bucket: Optional[int] = None, 121 bucket_max_pixels: Optional[int] = None,
122 progressive_buckets: bool = False, 122 progressive_buckets: bool = False,
123 dropout: float = 0, 123 dropout: float = 0,
124 interpolation: str = "bicubic", 124 interpolation: str = "bicubic",
@@ -145,7 +145,7 @@ class VlpnDataModule():
145 self.size = size 145 self.size = size
146 self.num_buckets = num_buckets 146 self.num_buckets = num_buckets
147 self.bucket_step_size = bucket_step_size 147 self.bucket_step_size = bucket_step_size
148 self.max_pixels_per_bucket = max_pixels_per_bucket 148 self.bucket_max_pixels = bucket_max_pixels
149 self.progressive_buckets = progressive_buckets 149 self.progressive_buckets = progressive_buckets
150 self.dropout = dropout 150 self.dropout = dropout
151 self.template_key = template_key 151 self.template_key = template_key
@@ -234,7 +234,7 @@ class VlpnDataModule():
234 train_dataset = VlpnDataset( 234 train_dataset = VlpnDataset(
235 self.data_train, self.prompt_processor, 235 self.data_train, self.prompt_processor,
236 num_buckets=self.num_buckets, progressive_buckets=self.progressive_buckets, 236 num_buckets=self.num_buckets, progressive_buckets=self.progressive_buckets,
237 bucket_step_size=self.bucket_step_size, max_pixels_per_bucket=self.max_pixels_per_bucket, 237 bucket_step_size=self.bucket_step_size, bucket_max_pixels=self.bucket_max_pixels,
238 batch_size=self.batch_size, generator=generator, 238 batch_size=self.batch_size, generator=generator,
239 size=self.size, interpolation=self.interpolation, 239 size=self.size, interpolation=self.interpolation,
240 num_class_images=self.num_class_images, dropout=self.dropout, shuffle=True, 240 num_class_images=self.num_class_images, dropout=self.dropout, shuffle=True,
@@ -264,7 +264,7 @@ class VlpnDataset(IterableDataset):
264 prompt_processor: PromptProcessor, 264 prompt_processor: PromptProcessor,
265 num_buckets: int = 1, 265 num_buckets: int = 1,
266 bucket_step_size: int = 64, 266 bucket_step_size: int = 64,
267 max_pixels_per_bucket: Optional[int] = None, 267 bucket_max_pixels: Optional[int] = None,
268 progressive_buckets: bool = False, 268 progressive_buckets: bool = False,
269 batch_size: int = 1, 269 batch_size: int = 1,
270 num_class_images: int = 0, 270 num_class_images: int = 0,
@@ -290,7 +290,7 @@ class VlpnDataset(IterableDataset):
290 base_size=size, 290 base_size=size,
291 step_size=bucket_step_size, 291 step_size=bucket_step_size,
292 num_buckets=num_buckets, 292 num_buckets=num_buckets,
293 max_pixels=max_pixels_per_bucket, 293 max_pixels=bucket_max_pixels,
294 progressive_buckets=progressive_buckets, 294 progressive_buckets=progressive_buckets,
295 ) 295 )
296 296