diff options
| author | Volpeon <git@volpeon.ink> | 2022-12-26 14:24:21 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-12-26 14:24:21 +0100 |
| commit | e0b686b475885f0c8480f7173eaa7359adf17e27 (patch) | |
| tree | 6ad882f152e63801d31230466e4d6468e7ada697 /data | |
| parent | Code simplifications, avoid autocast (diff) | |
| download | textual-inversion-diff-e0b686b475885f0c8480f7173eaa7359adf17e27.tar.gz textual-inversion-diff-e0b686b475885f0c8480f7173eaa7359adf17e27.tar.bz2 textual-inversion-diff-e0b686b475885f0c8480f7173eaa7359adf17e27.zip | |
Set default dimensions to 768; add config inheritance
Diffstat (limited to 'data')
| -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 0810c2c..0ad36dc 100644 --- a/data/csv.py +++ b/data/csv.py | |||
| @@ -51,7 +51,7 @@ class CSVDataModule(): | |||
| 51 | prompt_processor: PromptProcessor, | 51 | prompt_processor: PromptProcessor, |
| 52 | class_subdir: str = "cls", | 52 | class_subdir: str = "cls", |
| 53 | num_class_images: int = 1, | 53 | num_class_images: int = 1, |
| 54 | size: int = 512, | 54 | size: int = 768, |
| 55 | repeats: int = 1, | 55 | repeats: int = 1, |
| 56 | dropout: float = 0, | 56 | dropout: float = 0, |
| 57 | interpolation: str = "bicubic", | 57 | interpolation: str = "bicubic", |
| @@ -196,7 +196,7 @@ class CSVDataset(Dataset): | |||
| 196 | prompt_processor: PromptProcessor, | 196 | prompt_processor: PromptProcessor, |
| 197 | batch_size: int = 1, | 197 | batch_size: int = 1, |
| 198 | num_class_images: int = 0, | 198 | num_class_images: int = 0, |
| 199 | size: int = 512, | 199 | size: int = 768, |
| 200 | repeats: int = 1, | 200 | repeats: int = 1, |
| 201 | dropout: float = 0, | 201 | dropout: float = 0, |
| 202 | interpolation: str = "bicubic", | 202 | interpolation: str = "bicubic", |