summaryrefslogtreecommitdiffstats
path: root/data
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-05-05 10:51:14 +0200
committerVolpeon <git@volpeon.ink>2023-05-05 10:51:14 +0200
commit8d2aa65402c829583e26cdf2c336b8d3057657d6 (patch)
treecc2d47f56d1433e7600abd494361b1ae0a068f80 /data
parenttorch.compile won't work yet, keep code prepared (diff)
downloadtextual-inversion-diff-8d2aa65402c829583e26cdf2c336b8d3057657d6.tar.gz
textual-inversion-diff-8d2aa65402c829583e26cdf2c336b8d3057657d6.tar.bz2
textual-inversion-diff-8d2aa65402c829583e26cdf2c336b8d3057657d6.zip
Update
Diffstat (limited to 'data')
-rw-r--r--data/csv.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/data/csv.py b/data/csv.py
index c5e7aef..81e8b6b 100644
--- a/data/csv.py
+++ b/data/csv.py
@@ -187,7 +187,7 @@ class VlpnDataModule():
187 dropout: float = 0, 187 dropout: float = 0,
188 shuffle: bool = False, 188 shuffle: bool = False,
189 interpolation: str = "bicubic", 189 interpolation: str = "bicubic",
190 color_jitter: bool = True, 190 color_jitter: bool = False,
191 template_key: str = "template", 191 template_key: str = "template",
192 placeholder_tokens: list[str] = [], 192 placeholder_tokens: list[str] = [],
193 valid_set_size: Optional[int] = None, 193 valid_set_size: Optional[int] = None,
@@ -372,7 +372,7 @@ class VlpnDataset(IterableDataset):
372 dropout: float = 0, 372 dropout: float = 0,
373 shuffle: bool = False, 373 shuffle: bool = False,
374 interpolation: str = "bicubic", 374 interpolation: str = "bicubic",
375 color_jitter: bool = True, 375 color_jitter: bool = False,
376 generator: Optional[torch.Generator] = None, 376 generator: Optional[torch.Generator] = None,
377 npgenerator: Optional[np.random.Generator] = None, 377 npgenerator: Optional[np.random.Generator] = None,
378 ): 378 ):