diff options
| author | Volpeon <git@volpeon.ink> | 2023-05-05 10:51:14 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-05-05 10:51:14 +0200 |
| commit | 8d2aa65402c829583e26cdf2c336b8d3057657d6 (patch) | |
| tree | cc2d47f56d1433e7600abd494361b1ae0a068f80 /data/csv.py | |
| parent | torch.compile won't work yet, keep code prepared (diff) | |
| download | textual-inversion-diff-8d2aa65402c829583e26cdf2c336b8d3057657d6.tar.gz textual-inversion-diff-8d2aa65402c829583e26cdf2c336b8d3057657d6.tar.bz2 textual-inversion-diff-8d2aa65402c829583e26cdf2c336b8d3057657d6.zip | |
Update
Diffstat (limited to 'data/csv.py')
| -rw-r--r-- | data/csv.py | 4 |
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 | ): |