davidfant commited on
Commit
7a4ee0d
·
1 Parent(s): 7280f69

stable-diffusion-inpainting, one concept with specific prompt (2400 steps)

Browse files
args.json CHANGED
@@ -7,7 +7,7 @@
7
  "class_data_dir": null,
8
  "instance_prompt": null,
9
  "class_prompt": null,
10
- "save_sample_prompt": "photo of woman wearing sks bikini",
11
  "save_sample_negative_prompt": null,
12
  "n_save_sample": 4,
13
  "save_guidance_scale": 7.5,
@@ -23,8 +23,8 @@
23
  "train_text_encoder": true,
24
  "train_batch_size": 1,
25
  "sample_batch_size": 4,
26
- "num_train_epochs": 24,
27
- "max_train_steps": 7200,
28
  "gradient_accumulation_steps": 1,
29
  "gradient_checkpointing": false,
30
  "learning_rate": 1e-06,
@@ -42,152 +42,33 @@
42
  "hub_model_id": null,
43
  "logging_dir": "logs",
44
  "log_interval": 10,
45
- "save_interval": 2400,
46
  "save_min_steps": 0,
47
  "mixed_precision": "fp16",
48
  "not_cache_latents": true,
49
  "local_rank": -1,
50
  "concepts_list": [
51
- {
52
- "url": "https://www.brightswimwear.com/products/maui-top-flower-garden",
53
- "class_name": "bikini",
54
- "image_background": "pool and green bushes",
55
- "model": {
56
- "gender": "woman",
57
- "ethnicity": "white",
58
- "hair": "long dark brown"
59
- },
60
- "image_urls": [
61
- "https://cdn.shopify.com/s/files/1/2148/0245/products/1_72461194-d55a-4695-9e10-6e67f0834b7f.png?v=1627642383",
62
- "https://cdn.shopify.com/s/files/1/2148/0245/products/3_849aad81-ebf2-4925-ae13-2b2c3cf70c0a.png?v=1627642364",
63
- "https://cdn.shopify.com/s/files/1/2148/0245/products/2_fa361236-eb1c-462c-886d-5cb042ab311a.png?v=1627642364"
64
- ],
65
- "class_prompt": "a woman wearing a bikini",
66
- "class_data_dir": "./class/woman/bikini",
67
- "instance_name": "sks",
68
- "instance_prompt": "a white woman wearing a sks bikini, long dark brown hair, pool and green bushes background",
69
- "instance_data_dir": "./instance/sks",
70
- "class_mask_data_dir": "./class/woman/bikini-mask",
71
- "instance_mask_data_dir": "./instance/sks-mask"
72
- },
73
- {
74
- "url": "https://www.brightswimwear.com/products/tual-top-ribbed-green",
75
- "class_name": "bikini",
76
- "image_background": "beach",
77
- "model": {
78
- "gender": "woman",
79
- "ethnicity": "white",
80
- "hair": "long dark brown"
81
- },
82
- "image_urls": [
83
- "https://cdn.shopify.com/s/files/1/2148/0245/products/1_470afb11-8e2b-4d82-8121-4f4707cf7ea9.png?v=1643747143",
84
- "https://cdn.shopify.com/s/files/1/2148/0245/products/3_37328d6c-5819-4079-8891-0c9e899898a6.png?v=1643747146",
85
- "https://cdn.shopify.com/s/files/1/2148/0245/products/2_8d7e30e3-dc86-4f3c-a42c-aaa142f5588c.png?v=1643747146",
86
- "https://cdn.shopify.com/s/files/1/2148/0245/products/E-commerceProductImages_2_7e67be77-cb27-4f7b-a294-67486ba0ac99.png?v=1643747146"
87
- ],
88
- "class_prompt": "a woman wearing a bikini",
89
- "class_data_dir": "./class/woman/bikini",
90
- "instance_name": "rkx",
91
- "instance_prompt": "a white woman wearing a rkx bikini, long dark brown hair, beach background",
92
- "instance_data_dir": "./instance/rkx",
93
- "class_mask_data_dir": "./class/woman/bikini-mask",
94
- "instance_mask_data_dir": "./instance/rkx-mask"
95
- },
96
- {
97
- "url": "https://www.brightswimwear.com/products/tual-top-peach-stripe",
98
- "class_name": "bikini",
99
- "image_background": "beach",
100
- "model": {
101
- "gender": "woman",
102
- "ethnicity": "white",
103
- "hair": "long dark brown"
104
- },
105
- "image_urls": [
106
- "https://cdn.shopify.com/s/files/1/2148/0245/products/1_42d509f9-fa05-406f-b543-aced871a21fe.png?v=1644855913",
107
- "https://cdn.shopify.com/s/files/1/2148/0245/products/3_ecd3665a-3699-4ddd-9966-0cf33edf2bf7.png?v=1644855913",
108
- "https://cdn.shopify.com/s/files/1/2148/0245/products/2_04b9110a-a18f-4cb7-a794-66a677160e61.png?v=1644855913",
109
- "https://cdn.shopify.com/s/files/1/2148/0245/products/E-commerceProductImages_4.png?v=1644855913",
110
- "https://cdn.shopify.com/s/files/1/2148/0245/products/4_87598809-81b5-4284-b612-3ae9ccb41149.png?v=1644855868"
111
- ],
112
- "class_prompt": "a woman wearing a bikini",
113
- "class_data_dir": "./class/woman/bikini",
114
- "instance_name": "ptf",
115
- "instance_prompt": "a white woman wearing a ptf bikini, long dark brown hair, beach background",
116
- "instance_data_dir": "./instance/ptf",
117
- "class_mask_data_dir": "./class/woman/bikini-mask",
118
- "instance_mask_data_dir": "./instance/ptf-mask"
119
- },
120
  {
121
  "url": "https://www.brightswimwear.com/products/bali-top-ocean-dream",
122
- "class_name": "bikini",
123
- "model": {
124
- "gender": "woman",
125
- "ethnicity": "white",
126
- "hair": "long dark brown"
127
- },
128
  "image_urls": [
129
  "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1658.jpg?v=1666613045",
130
  "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1664.jpg?v=1666613045",
131
- "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1671.jpg?v=1666613045",
132
- "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1680.jpg?v=1666613045",
133
- "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-296.jpg?v=1666613045",
134
- "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-297.jpg?v=1666613045",
135
- "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-299.jpg?v=1666613045",
136
- "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-289.jpg?v=1666613045",
137
- "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-298.jpg?v=1666613042"
138
  ],
 
139
  "class_prompt": "a woman wearing a bikini",
140
  "class_data_dir": "./class/woman/bikini",
141
  "instance_name": "qwe",
142
- "instance_prompt": "a white woman wearing a qwe bikini, long dark brown hair",
143
  "instance_data_dir": "./instance/qwe",
144
  "class_mask_data_dir": "./class/woman/bikini-mask",
145
  "instance_mask_data_dir": "./instance/qwe-mask"
146
- },
147
- {
148
- "url": "https://www.brightswimwear.com/products/bali-top-lime",
149
- "class_name": "bikini",
150
- "model": {
151
- "gender": "woman",
152
- "ethnicity": "white",
153
- "hair": "long dark brown"
154
- },
155
- "image_urls": [
156
- "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC0869.jpg?v=1654035999",
157
- "https://cdn.shopify.com/s/files/1/2148/0245/products/Bright_studio-442.jpg?v=1654035999",
158
- "https://cdn.shopify.com/s/files/1/2148/0245/products/Bright_studio-458.jpg?v=1654035999",
159
- "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC0818.jpg?v=1654035999"
160
- ],
161
- "class_prompt": "a woman wearing a bikini",
162
- "class_data_dir": "./class/woman/bikini",
163
- "instance_name": "kxy",
164
- "instance_prompt": "a white woman wearing a kxy bikini, long dark brown hair",
165
- "instance_data_dir": "./instance/kxy",
166
- "class_mask_data_dir": "./class/woman/bikini-mask",
167
- "instance_mask_data_dir": "./instance/kxy-mask"
168
- },
169
- {
170
- "url": "https://www.brightswimwear.com/products/maria-top-lime",
171
- "class_name": "bikini",
172
- "model": {
173
- "gender": "woman",
174
- "ethnicity": "white",
175
- "hair": "long dark brown"
176
- },
177
- "image_urls": [
178
- "https://cdn.shopify.com/s/files/1/2148/0245/products/E-COMMERCEIMAGES-x3_1400x2100px_5.jpg?v=1654025565",
179
- "https://cdn.shopify.com/s/files/1/2148/0245/products/Bright_studio-085.jpg?v=1654025565",
180
- "https://cdn.shopify.com/s/files/1/2148/0245/products/Bright_studio-077.jpg?v=1654025565",
181
- "https://cdn.shopify.com/s/files/1/2148/0245/products/Bright_studio-089.jpg?v=1654025565",
182
- "https://cdn.shopify.com/s/files/1/2148/0245/products/E-COMMERCEIMAGES-x3_1400x2100px_4.jpg?v=1654025565"
183
- ],
184
- "class_prompt": "a woman wearing a bikini",
185
- "class_data_dir": "./class/woman/bikini",
186
- "instance_name": "yzu",
187
- "instance_prompt": "a white woman wearing a yzu bikini, long dark brown hair",
188
- "instance_data_dir": "./instance/yzu",
189
- "class_mask_data_dir": "./class/woman/bikini-mask",
190
- "instance_mask_data_dir": "./instance/yzu-mask"
191
  }
192
  ]
193
  }
 
7
  "class_data_dir": null,
8
  "instance_prompt": null,
9
  "class_prompt": null,
10
+ "save_sample_prompt": "a woman wearing a blue qwe bikini with cyan floral patterns, short blonde hair, on a boat with an island in the background",
11
  "save_sample_negative_prompt": null,
12
  "n_save_sample": 4,
13
  "save_guidance_scale": 7.5,
 
23
  "train_text_encoder": true,
24
  "train_batch_size": 1,
25
  "sample_batch_size": 4,
26
+ "num_train_epochs": 200000,
27
+ "max_train_steps": 9999999,
28
  "gradient_accumulation_steps": 1,
29
  "gradient_checkpointing": false,
30
  "learning_rate": 1e-06,
 
42
  "hub_model_id": null,
43
  "logging_dir": "logs",
44
  "log_interval": 10,
45
+ "save_interval": 800,
46
  "save_min_steps": 0,
47
  "mixed_precision": "fp16",
48
  "not_cache_latents": true,
49
  "local_rank": -1,
50
  "concepts_list": [
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
  {
52
  "url": "https://www.brightswimwear.com/products/bali-top-ocean-dream",
 
 
 
 
 
 
53
  "image_urls": [
54
  "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1658.jpg?v=1666613045",
55
  "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1664.jpg?v=1666613045",
56
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1671.jpg?v=1666613045&width=1946",
57
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1680.jpg?v=1666613045&width=1946",
58
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-296.jpg?v=1666613045&width=1946",
59
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-297.jpg?v=1666613045&width=1946",
60
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-299.jpg?v=1666613045&width=1946",
61
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-289.jpg?v=1666613045&width=1946",
62
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-298.jpg?v=1666613042&width=1946"
63
  ],
64
+ "class_name": "bikini",
65
  "class_prompt": "a woman wearing a bikini",
66
  "class_data_dir": "./class/woman/bikini",
67
  "instance_name": "qwe",
68
+ "instance_prompt": "a white woman wearing a blue qwe bikini with cyan floral patterns, long dark brown hair",
69
  "instance_data_dir": "./instance/qwe",
70
  "class_mask_data_dir": "./class/woman/bikini-mask",
71
  "instance_mask_data_dir": "./instance/qwe-mask"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
72
  }
73
  ]
74
  }
samples/0.png CHANGED
samples/1.png CHANGED
samples/2.png CHANGED
samples/3.png CHANGED
text_encoder/pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8238a78c3f33f73344b7d9a382a4ab95249971bc2aa63393494db6a6b32b559d
3
  size 492308087
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02c7a24bf5ca946689727de3a6627ef50af9ac0ea63b976f746ec0cf905600c3
3
  size 492308087
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d4f021b0967405f639140680bcbcceeaf690b60c353ea2f122fa8ea2d70e0e67
3
  size 3438421925
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90c42317349565625291ae740359000c8d62144b6b665413e832151496ec642e
3
  size 3438421925