diff options
| author | alg-wiki <alg.4chan@gmail.com> | 2022-10-10 13:35:35 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-10 14:05:12 +0000 |
| commit | 7a20f914eddfdf09c0ccced157ec108205bc3d0f (patch) | |
| tree | 5a04127ec59ad70f7c3050b903e480964e3866b6 /modules/textual_inversion/dataset.py | |
| parent | 6ad3a53e368d36535de1a4fca73b3bb78fd40654 (diff) | |
| download | stable-diffusion-webui-gfx803-7a20f914eddfdf09c0ccced157ec108205bc3d0f.tar.gz stable-diffusion-webui-gfx803-7a20f914eddfdf09c0ccced157ec108205bc3d0f.tar.bz2 stable-diffusion-webui-gfx803-7a20f914eddfdf09c0ccced157ec108205bc3d0f.zip | |
Custom Width and Height
Diffstat (limited to 'modules/textual_inversion/dataset.py')
| -rw-r--r-- | modules/textual_inversion/dataset.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/modules/textual_inversion/dataset.py b/modules/textual_inversion/dataset.py index acc4ce59..bcf772d2 100644 --- a/modules/textual_inversion/dataset.py +++ b/modules/textual_inversion/dataset.py @@ -15,13 +15,12 @@ re_tag = re.compile(r"[a-zA-Z][_\w\d()]+") class PersonalizedBase(Dataset):
- def __init__(self, data_root, size, repeats, flip_p=0.5, placeholder_token="*", model=None, device=None, template_file=None):
+ def __init__(self, data_root, width, height, repeats, flip_p=0.5, placeholder_token="*", model=None, device=None, template_file=None):
self.placeholder_token = placeholder_token
- self.size = size
- self.width = size
- self.height = size
+ self.width = width
+ self.height = height
self.flip = transforms.RandomHorizontalFlip(p=flip_p)
self.dataset = []
|