Wi-zz commited on
Commit
81e854a
·
verified ·
1 Parent(s): 737aa50

Upload folder using huggingface_hub

Browse files
ZIB/config/.ipynb_checkpoints/reddit-2-zib2026-02-15_23-12-50-checkpoint.json ADDED
@@ -0,0 +1,1566 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version": 10,
3
+ "training_method": "LORA",
4
+ "model_type": "Z_IMAGE",
5
+ "debug_mode": false,
6
+ "debug_dir": "debug",
7
+ "workspace_dir": "workspace/run",
8
+ "cache_dir": "workspace-cache/run",
9
+ "tensorboard": true,
10
+ "tensorboard_expose": false,
11
+ "tensorboard_always_on": false,
12
+ "tensorboard_port": 6007,
13
+ "validation": false,
14
+ "validate_after": 1,
15
+ "validate_after_unit": "EPOCH",
16
+ "continue_last_backup": false,
17
+ "include_train_config": "NONE",
18
+ "multi_gpu": true,
19
+ "device_indexes": "0,1",
20
+ "gradient_reduce_precision": "FLOAT_32_STOCHASTIC",
21
+ "fused_gradient_reduce": true,
22
+ "async_gradient_reduce": true,
23
+ "async_gradient_reduce_buffer": 100,
24
+ "base_model_name": "Tongyi-MAI/Z-Image",
25
+ "output_dtype": "BFLOAT_16",
26
+ "output_model_format": "SAFETENSORS",
27
+ "output_model_destination": "models/reddit-2-zib.safetensors",
28
+ "gradient_checkpointing": "ON",
29
+ "enable_async_offloading": true,
30
+ "enable_activation_offloading": true,
31
+ "layer_offload_fraction": 0.001,
32
+ "force_circular_padding": false,
33
+ "compile": true,
34
+ "concept_file_name": "training_concepts/concepts.json",
35
+ "concepts": [
36
+ {
37
+ "__version": 2,
38
+ "image": {
39
+ "__version": 0,
40
+ "enable_crop_jitter": true,
41
+ "enable_random_flip": false,
42
+ "enable_fixed_flip": false,
43
+ "enable_random_rotate": false,
44
+ "enable_fixed_rotate": false,
45
+ "random_rotate_max_angle": 0.0,
46
+ "enable_random_brightness": false,
47
+ "enable_fixed_brightness": false,
48
+ "random_brightness_max_strength": 0.0,
49
+ "enable_random_contrast": false,
50
+ "enable_fixed_contrast": false,
51
+ "random_contrast_max_strength": 0.0,
52
+ "enable_random_saturation": false,
53
+ "enable_fixed_saturation": false,
54
+ "random_saturation_max_strength": 0.0,
55
+ "enable_random_hue": false,
56
+ "enable_fixed_hue": false,
57
+ "random_hue_max_strength": 0.0,
58
+ "enable_resolution_override": true,
59
+ "resolution_override": "1024",
60
+ "enable_random_circular_mask_shrink": false,
61
+ "enable_random_mask_rotate_crop": false
62
+ },
63
+ "text": {
64
+ "__version": 0,
65
+ "prompt_source": "sample",
66
+ "prompt_path": "",
67
+ "enable_tag_shuffling": false,
68
+ "tag_delimiter": ",",
69
+ "keep_tags_count": 1,
70
+ "tag_dropout_enable": false,
71
+ "tag_dropout_mode": "FULL",
72
+ "tag_dropout_probability": 0.0,
73
+ "tag_dropout_special_tags_mode": "NONE",
74
+ "tag_dropout_special_tags": "",
75
+ "tag_dropout_special_tags_regex": false,
76
+ "caps_randomize_enable": false,
77
+ "caps_randomize_mode": "capslock, title, first, random",
78
+ "caps_randomize_probability": 0.0,
79
+ "caps_randomize_lowercase": false
80
+ },
81
+ "name": "cock-1024",
82
+ "path": "/workspace/Datasets/r_cock/1408",
83
+ "seed": -151055228,
84
+ "enabled": true,
85
+ "type": "STANDARD",
86
+ "include_subdirectories": false,
87
+ "image_variations": 1,
88
+ "text_variations": 1,
89
+ "balancing": 1.0,
90
+ "balancing_strategy": "REPEATS",
91
+ "loss_weight": 1.0,
92
+ "concept_stats": {
93
+ "file_size": 18979799,
94
+ "image_count": 84,
95
+ "image_with_mask_count": 0,
96
+ "image_with_caption_count": 84,
97
+ "video_count": 0,
98
+ "video_with_mask_count": 0,
99
+ "video_with_caption_count": 0,
100
+ "mask_count": 0,
101
+ "paired_masks": 0,
102
+ "unpaired_masks": 0,
103
+ "caption_count": 84,
104
+ "subcaption_count": 468,
105
+ "paired_captions": 84,
106
+ "unpaired_captions": 0,
107
+ "processing_time": 0.022084553027525544,
108
+ "directory_count": 1,
109
+ "max_pixels": [
110
+ 1972080,
111
+ "cock_2022-12-11_03-13-22.jpg",
112
+ "1080w x 1826h"
113
+ ],
114
+ "min_pixels": [
115
+ 1065636,
116
+ "cock_2026-01-10_08-01-31_03.webp",
117
+ "1242w x 858h"
118
+ ],
119
+ "avg_pixels": 1498911.7261904762,
120
+ "max_length": [
121
+ 0,
122
+ "-"
123
+ ],
124
+ "min_length": [
125
+ 1000000000,
126
+ "-"
127
+ ],
128
+ "avg_length": 0,
129
+ "max_fps": [
130
+ 0,
131
+ "-"
132
+ ],
133
+ "min_fps": [
134
+ 1000000000,
135
+ "-"
136
+ ],
137
+ "avg_fps": 0,
138
+ "max_caption_length": [
139
+ 892,
140
+ "cock_2026-02-13_06-20-19.jpg",
141
+ 136
142
+ ],
143
+ "min_caption_length": [
144
+ 0,
145
+ "cock_2019-07-24_12-02-42.jpg",
146
+ 0
147
+ ],
148
+ "avg_caption_length": [
149
+ 306.34670873400086,
150
+ 50.15109395168455
151
+ ],
152
+ "aspect_buckets": {
153
+ "0.25": 0,
154
+ "0.285714": 0,
155
+ "0.333333": 0,
156
+ "0.4": 0,
157
+ "0.5": 3,
158
+ "0.571429": 0,
159
+ "0.666667": 1,
160
+ "0.8": 2,
161
+ "1.0": 4,
162
+ "1.25": 43,
163
+ "1.5": 9,
164
+ "1.75": 17,
165
+ "2.0": 5,
166
+ "2.5": 0,
167
+ "3.0": 0,
168
+ "3.5": 0,
169
+ "4.0": 0
170
+ },
171
+ "force_cancelled": false
172
+ }
173
+ },
174
+ {
175
+ "__version": 2,
176
+ "image": {
177
+ "__version": 0,
178
+ "enable_crop_jitter": true,
179
+ "enable_random_flip": false,
180
+ "enable_fixed_flip": false,
181
+ "enable_random_rotate": false,
182
+ "enable_fixed_rotate": false,
183
+ "random_rotate_max_angle": 0.0,
184
+ "enable_random_brightness": false,
185
+ "enable_fixed_brightness": false,
186
+ "random_brightness_max_strength": 0.0,
187
+ "enable_random_contrast": false,
188
+ "enable_fixed_contrast": false,
189
+ "random_contrast_max_strength": 0.0,
190
+ "enable_random_saturation": false,
191
+ "enable_fixed_saturation": false,
192
+ "random_saturation_max_strength": 0.0,
193
+ "enable_random_hue": false,
194
+ "enable_fixed_hue": false,
195
+ "random_hue_max_strength": 0.0,
196
+ "enable_resolution_override": false,
197
+ "resolution_override": "1024,1536",
198
+ "enable_random_circular_mask_shrink": false,
199
+ "enable_random_mask_rotate_crop": false
200
+ },
201
+ "text": {
202
+ "__version": 0,
203
+ "prompt_source": "sample",
204
+ "prompt_path": "",
205
+ "enable_tag_shuffling": false,
206
+ "tag_delimiter": ",",
207
+ "keep_tags_count": 1,
208
+ "tag_dropout_enable": false,
209
+ "tag_dropout_mode": "FULL",
210
+ "tag_dropout_probability": 0.0,
211
+ "tag_dropout_special_tags_mode": "NONE",
212
+ "tag_dropout_special_tags": "",
213
+ "tag_dropout_special_tags_regex": false,
214
+ "caps_randomize_enable": false,
215
+ "caps_randomize_mode": "capslock, title, first, random",
216
+ "caps_randomize_probability": 0.0,
217
+ "caps_randomize_lowercase": false
218
+ },
219
+ "name": "cock-1536",
220
+ "path": "/workspace/Datasets/r_cock/1792",
221
+ "seed": 857416178,
222
+ "enabled": true,
223
+ "type": "STANDARD",
224
+ "include_subdirectories": false,
225
+ "image_variations": 1,
226
+ "text_variations": 1,
227
+ "balancing": 1.0,
228
+ "balancing_strategy": "REPEATS",
229
+ "loss_weight": 1.0,
230
+ "concept_stats": {
231
+ "file_size": 26015793,
232
+ "image_count": 61,
233
+ "image_with_mask_count": 0,
234
+ "image_with_caption_count": 61,
235
+ "video_count": 0,
236
+ "video_with_mask_count": 0,
237
+ "video_with_caption_count": 0,
238
+ "mask_count": 0,
239
+ "paired_masks": 0,
240
+ "unpaired_masks": 0,
241
+ "caption_count": 61,
242
+ "subcaption_count": 333,
243
+ "paired_captions": 61,
244
+ "unpaired_captions": 0,
245
+ "processing_time": 0.016968363896012306,
246
+ "directory_count": 1,
247
+ "max_pixels": [
248
+ 3165170,
249
+ "cock_2020-06-22_12-19-13.jpg",
250
+ "1535w x 2062h"
251
+ ],
252
+ "min_pixels": [
253
+ 1984940,
254
+ "cock_2026-02-13_18-10-26.jpg",
255
+ "1220w x 1627h"
256
+ ],
257
+ "avg_pixels": 2598588.032786885,
258
+ "max_length": [
259
+ 0,
260
+ "-"
261
+ ],
262
+ "min_length": [
263
+ 1000000000,
264
+ "-"
265
+ ],
266
+ "avg_length": 0,
267
+ "max_fps": [
268
+ 0,
269
+ "-"
270
+ ],
271
+ "min_fps": [
272
+ 1000000000,
273
+ "-"
274
+ ],
275
+ "avg_fps": 0,
276
+ "max_caption_length": [
277
+ 811,
278
+ "cock_2025-11-26_13-35-55_01.webp",
279
+ 135
280
+ ],
281
+ "min_caption_length": [
282
+ 0,
283
+ "cock_2019-02-05_08-32-21.jpg",
284
+ 0
285
+ ],
286
+ "avg_caption_length": [
287
+ 311.8868407765503,
288
+ 50.57262830257678
289
+ ],
290
+ "aspect_buckets": {
291
+ "0.25": 0,
292
+ "0.285714": 0,
293
+ "0.333333": 0,
294
+ "0.4": 0,
295
+ "0.5": 0,
296
+ "0.571429": 0,
297
+ "0.666667": 1,
298
+ "0.8": 3,
299
+ "1.0": 2,
300
+ "1.25": 16,
301
+ "1.5": 7,
302
+ "1.75": 27,
303
+ "2.0": 5,
304
+ "2.5": 0,
305
+ "3.0": 0,
306
+ "3.5": 0,
307
+ "4.0": 0
308
+ },
309
+ "force_cancelled": false
310
+ }
311
+ },
312
+ {
313
+ "__version": 2,
314
+ "image": {
315
+ "__version": 0,
316
+ "enable_crop_jitter": true,
317
+ "enable_random_flip": false,
318
+ "enable_fixed_flip": false,
319
+ "enable_random_rotate": false,
320
+ "enable_fixed_rotate": false,
321
+ "random_rotate_max_angle": 0.0,
322
+ "enable_random_brightness": false,
323
+ "enable_fixed_brightness": false,
324
+ "random_brightness_max_strength": 0.0,
325
+ "enable_random_contrast": false,
326
+ "enable_fixed_contrast": false,
327
+ "random_contrast_max_strength": 0.0,
328
+ "enable_random_saturation": false,
329
+ "enable_fixed_saturation": false,
330
+ "random_saturation_max_strength": 0.0,
331
+ "enable_random_hue": false,
332
+ "enable_fixed_hue": false,
333
+ "random_hue_max_strength": 0.0,
334
+ "enable_resolution_override": false,
335
+ "resolution_override": "512",
336
+ "enable_random_circular_mask_shrink": false,
337
+ "enable_random_mask_rotate_crop": false
338
+ },
339
+ "text": {
340
+ "__version": 0,
341
+ "prompt_source": "sample",
342
+ "prompt_path": "",
343
+ "enable_tag_shuffling": false,
344
+ "tag_delimiter": ",",
345
+ "keep_tags_count": 1,
346
+ "tag_dropout_enable": false,
347
+ "tag_dropout_mode": "FULL",
348
+ "tag_dropout_probability": 0.0,
349
+ "tag_dropout_special_tags_mode": "NONE",
350
+ "tag_dropout_special_tags": "",
351
+ "tag_dropout_special_tags_regex": false,
352
+ "caps_randomize_enable": false,
353
+ "caps_randomize_mode": "capslock, title, first, random",
354
+ "caps_randomize_probability": 0.0,
355
+ "caps_randomize_lowercase": false
356
+ },
357
+ "name": "cock-2048",
358
+ "path": "/workspace/Datasets/r_cock/2048",
359
+ "seed": -313616135,
360
+ "enabled": true,
361
+ "type": "STANDARD",
362
+ "include_subdirectories": false,
363
+ "image_variations": 1,
364
+ "text_variations": 1,
365
+ "balancing": 1.0,
366
+ "balancing_strategy": "REPEATS",
367
+ "loss_weight": 1.0,
368
+ "concept_stats": {
369
+ "file_size": 413667396,
370
+ "image_count": 351,
371
+ "image_with_mask_count": 0,
372
+ "image_with_caption_count": 351,
373
+ "video_count": 0,
374
+ "video_with_mask_count": 0,
375
+ "video_with_caption_count": 0,
376
+ "mask_count": 0,
377
+ "paired_masks": 0,
378
+ "unpaired_masks": 0,
379
+ "caption_count": 351,
380
+ "subcaption_count": 1941,
381
+ "paired_captions": 351,
382
+ "unpaired_captions": 0,
383
+ "processing_time": 0.05115061393007636,
384
+ "directory_count": 1,
385
+ "max_pixels": [
386
+ 49939200,
387
+ "cock_2026-02-12_19-39-44.jpg",
388
+ "6120w x 8160h"
389
+ ],
390
+ "min_pixels": [
391
+ 3232800,
392
+ "cock_2019-12-26_09-52-24.png",
393
+ "1440w x 2245h"
394
+ ],
395
+ "avg_pixels": 9450439.341880342,
396
+ "max_length": [
397
+ 0,
398
+ "-"
399
+ ],
400
+ "min_length": [
401
+ 1000000000,
402
+ "-"
403
+ ],
404
+ "avg_length": 0,
405
+ "max_fps": [
406
+ 0,
407
+ "-"
408
+ ],
409
+ "min_fps": [
410
+ 1000000000,
411
+ "-"
412
+ ],
413
+ "avg_fps": 0,
414
+ "max_caption_length": [
415
+ 913,
416
+ "cock_2025-02-23_16-15-28_03.webp",
417
+ 145
418
+ ],
419
+ "min_caption_length": [
420
+ 0,
421
+ "cock_2019-02-20_17-33-55.jpg",
422
+ 0
423
+ ],
424
+ "avg_caption_length": [
425
+ 308.8869282186318,
426
+ 50.642955884895436
427
+ ],
428
+ "aspect_buckets": {
429
+ "0.25": 0,
430
+ "0.285714": 0,
431
+ "0.333333": 0,
432
+ "0.4": 1,
433
+ "0.5": 0,
434
+ "0.571429": 6,
435
+ "0.666667": 3,
436
+ "0.8": 54,
437
+ "1.0": 17,
438
+ "1.25": 219,
439
+ "1.5": 23,
440
+ "1.75": 18,
441
+ "2.0": 9,
442
+ "2.5": 0,
443
+ "3.0": 1,
444
+ "3.5": 0,
445
+ "4.0": 0
446
+ },
447
+ "force_cancelled": false
448
+ }
449
+ },
450
+ {
451
+ "__version": 2,
452
+ "image": {
453
+ "__version": 0,
454
+ "enable_crop_jitter": true,
455
+ "enable_random_flip": false,
456
+ "enable_fixed_flip": false,
457
+ "enable_random_rotate": false,
458
+ "enable_fixed_rotate": false,
459
+ "random_rotate_max_angle": 0.0,
460
+ "enable_random_brightness": false,
461
+ "enable_fixed_brightness": false,
462
+ "random_brightness_max_strength": 0.0,
463
+ "enable_random_contrast": false,
464
+ "enable_fixed_contrast": false,
465
+ "random_contrast_max_strength": 0.0,
466
+ "enable_random_saturation": false,
467
+ "enable_fixed_saturation": false,
468
+ "random_saturation_max_strength": 0.0,
469
+ "enable_random_hue": false,
470
+ "enable_fixed_hue": false,
471
+ "random_hue_max_strength": 0.0,
472
+ "enable_resolution_override": true,
473
+ "resolution_override": "1024",
474
+ "enable_random_circular_mask_shrink": false,
475
+ "enable_random_mask_rotate_crop": false
476
+ },
477
+ "text": {
478
+ "__version": 0,
479
+ "prompt_source": "sample",
480
+ "prompt_path": "",
481
+ "enable_tag_shuffling": false,
482
+ "tag_delimiter": ",",
483
+ "keep_tags_count": 1,
484
+ "tag_dropout_enable": false,
485
+ "tag_dropout_mode": "FULL",
486
+ "tag_dropout_probability": 0.0,
487
+ "tag_dropout_special_tags_mode": "NONE",
488
+ "tag_dropout_special_tags": "",
489
+ "tag_dropout_special_tags_regex": false,
490
+ "caps_randomize_enable": false,
491
+ "caps_randomize_mode": "capslock, title, first, random",
492
+ "caps_randomize_probability": 0.0,
493
+ "caps_randomize_lowercase": false
494
+ },
495
+ "name": "gayfaceandcock-1024",
496
+ "path": "/workspace/Datasets/r_gayfaceandcock/1384",
497
+ "seed": -51838343,
498
+ "enabled": true,
499
+ "type": "STANDARD",
500
+ "include_subdirectories": false,
501
+ "image_variations": 1,
502
+ "text_variations": 1,
503
+ "balancing": 1.0,
504
+ "balancing_strategy": "REPEATS",
505
+ "loss_weight": 1.0,
506
+ "concept_stats": {}
507
+ },
508
+ {
509
+ "__version": 2,
510
+ "image": {
511
+ "__version": 0,
512
+ "enable_crop_jitter": true,
513
+ "enable_random_flip": false,
514
+ "enable_fixed_flip": false,
515
+ "enable_random_rotate": false,
516
+ "enable_fixed_rotate": false,
517
+ "random_rotate_max_angle": 0.0,
518
+ "enable_random_brightness": false,
519
+ "enable_fixed_brightness": false,
520
+ "random_brightness_max_strength": 0.0,
521
+ "enable_random_contrast": false,
522
+ "enable_fixed_contrast": false,
523
+ "random_contrast_max_strength": 0.0,
524
+ "enable_random_saturation": false,
525
+ "enable_fixed_saturation": false,
526
+ "random_saturation_max_strength": 0.0,
527
+ "enable_random_hue": false,
528
+ "enable_fixed_hue": false,
529
+ "random_hue_max_strength": 0.0,
530
+ "enable_resolution_override": false,
531
+ "resolution_override": "512",
532
+ "enable_random_circular_mask_shrink": false,
533
+ "enable_random_mask_rotate_crop": false
534
+ },
535
+ "text": {
536
+ "__version": 0,
537
+ "prompt_source": "sample",
538
+ "prompt_path": "",
539
+ "enable_tag_shuffling": false,
540
+ "tag_delimiter": ",",
541
+ "keep_tags_count": 1,
542
+ "tag_dropout_enable": false,
543
+ "tag_dropout_mode": "FULL",
544
+ "tag_dropout_probability": 0.0,
545
+ "tag_dropout_special_tags_mode": "NONE",
546
+ "tag_dropout_special_tags": "",
547
+ "tag_dropout_special_tags_regex": false,
548
+ "caps_randomize_enable": false,
549
+ "caps_randomize_mode": "capslock, title, first, random",
550
+ "caps_randomize_probability": 0.0,
551
+ "caps_randomize_lowercase": false
552
+ },
553
+ "name": "gayfaceandcock-1536",
554
+ "path": "/workspace/Datasets/r_gayfaceandcock/1792",
555
+ "seed": -461368761,
556
+ "enabled": true,
557
+ "type": "STANDARD",
558
+ "include_subdirectories": false,
559
+ "image_variations": 1,
560
+ "text_variations": 1,
561
+ "balancing": 1.0,
562
+ "balancing_strategy": "REPEATS",
563
+ "loss_weight": 1.0,
564
+ "concept_stats": {}
565
+ },
566
+ {
567
+ "__version": 2,
568
+ "image": {
569
+ "__version": 0,
570
+ "enable_crop_jitter": true,
571
+ "enable_random_flip": false,
572
+ "enable_fixed_flip": false,
573
+ "enable_random_rotate": false,
574
+ "enable_fixed_rotate": false,
575
+ "random_rotate_max_angle": 0.0,
576
+ "enable_random_brightness": false,
577
+ "enable_fixed_brightness": false,
578
+ "random_brightness_max_strength": 0.0,
579
+ "enable_random_contrast": false,
580
+ "enable_fixed_contrast": false,
581
+ "random_contrast_max_strength": 0.0,
582
+ "enable_random_saturation": false,
583
+ "enable_fixed_saturation": false,
584
+ "random_saturation_max_strength": 0.0,
585
+ "enable_random_hue": false,
586
+ "enable_fixed_hue": false,
587
+ "random_hue_max_strength": 0.0,
588
+ "enable_resolution_override": false,
589
+ "resolution_override": "512",
590
+ "enable_random_circular_mask_shrink": false,
591
+ "enable_random_mask_rotate_crop": false
592
+ },
593
+ "text": {
594
+ "__version": 0,
595
+ "prompt_source": "sample",
596
+ "prompt_path": "",
597
+ "enable_tag_shuffling": false,
598
+ "tag_delimiter": ",",
599
+ "keep_tags_count": 1,
600
+ "tag_dropout_enable": false,
601
+ "tag_dropout_mode": "FULL",
602
+ "tag_dropout_probability": 0.0,
603
+ "tag_dropout_special_tags_mode": "NONE",
604
+ "tag_dropout_special_tags": "",
605
+ "tag_dropout_special_tags_regex": false,
606
+ "caps_randomize_enable": false,
607
+ "caps_randomize_mode": "capslock, title, first, random",
608
+ "caps_randomize_probability": 0.0,
609
+ "caps_randomize_lowercase": false
610
+ },
611
+ "name": "gayfaceandcock-2048",
612
+ "path": "/workspace/Datasets/r_gayfaceandcock/2048",
613
+ "seed": 634934184,
614
+ "enabled": true,
615
+ "type": "STANDARD",
616
+ "include_subdirectories": false,
617
+ "image_variations": 1,
618
+ "text_variations": 1,
619
+ "balancing": 1.0,
620
+ "balancing_strategy": "REPEATS",
621
+ "loss_weight": 1.0,
622
+ "concept_stats": {}
623
+ },
624
+ {
625
+ "__version": 2,
626
+ "image": {
627
+ "__version": 0,
628
+ "enable_crop_jitter": true,
629
+ "enable_random_flip": false,
630
+ "enable_fixed_flip": false,
631
+ "enable_random_rotate": false,
632
+ "enable_fixed_rotate": false,
633
+ "random_rotate_max_angle": 0.0,
634
+ "enable_random_brightness": false,
635
+ "enable_fixed_brightness": false,
636
+ "random_brightness_max_strength": 0.0,
637
+ "enable_random_contrast": false,
638
+ "enable_fixed_contrast": false,
639
+ "random_contrast_max_strength": 0.0,
640
+ "enable_random_saturation": false,
641
+ "enable_fixed_saturation": false,
642
+ "random_saturation_max_strength": 0.0,
643
+ "enable_random_hue": false,
644
+ "enable_fixed_hue": false,
645
+ "random_hue_max_strength": 0.0,
646
+ "enable_resolution_override": true,
647
+ "resolution_override": "1024",
648
+ "enable_random_circular_mask_shrink": false,
649
+ "enable_random_mask_rotate_crop": false
650
+ },
651
+ "text": {
652
+ "__version": 0,
653
+ "prompt_source": "sample",
654
+ "prompt_path": "",
655
+ "enable_tag_shuffling": false,
656
+ "tag_delimiter": ",",
657
+ "keep_tags_count": 1,
658
+ "tag_dropout_enable": false,
659
+ "tag_dropout_mode": "FULL",
660
+ "tag_dropout_probability": 0.0,
661
+ "tag_dropout_special_tags_mode": "NONE",
662
+ "tag_dropout_special_tags": "",
663
+ "tag_dropout_special_tags_regex": false,
664
+ "caps_randomize_enable": false,
665
+ "caps_randomize_mode": "capslock, title, first, random",
666
+ "caps_randomize_probability": 0.0,
667
+ "caps_randomize_lowercase": false
668
+ },
669
+ "name": "ladybonersgw-1024",
670
+ "path": "/workspace/Datasets/r_ladybonersgw/1408",
671
+ "seed": -39295866,
672
+ "enabled": true,
673
+ "type": "STANDARD",
674
+ "include_subdirectories": false,
675
+ "image_variations": 1,
676
+ "text_variations": 1,
677
+ "balancing": 1.0,
678
+ "balancing_strategy": "REPEATS",
679
+ "loss_weight": 1.0,
680
+ "concept_stats": {}
681
+ },
682
+ {
683
+ "__version": 2,
684
+ "image": {
685
+ "__version": 0,
686
+ "enable_crop_jitter": true,
687
+ "enable_random_flip": false,
688
+ "enable_fixed_flip": false,
689
+ "enable_random_rotate": false,
690
+ "enable_fixed_rotate": false,
691
+ "random_rotate_max_angle": 0.0,
692
+ "enable_random_brightness": false,
693
+ "enable_fixed_brightness": false,
694
+ "random_brightness_max_strength": 0.0,
695
+ "enable_random_contrast": false,
696
+ "enable_fixed_contrast": false,
697
+ "random_contrast_max_strength": 0.0,
698
+ "enable_random_saturation": false,
699
+ "enable_fixed_saturation": false,
700
+ "random_saturation_max_strength": 0.0,
701
+ "enable_random_hue": false,
702
+ "enable_fixed_hue": false,
703
+ "random_hue_max_strength": 0.0,
704
+ "enable_resolution_override": false,
705
+ "resolution_override": "512",
706
+ "enable_random_circular_mask_shrink": false,
707
+ "enable_random_mask_rotate_crop": false
708
+ },
709
+ "text": {
710
+ "__version": 0,
711
+ "prompt_source": "sample",
712
+ "prompt_path": "",
713
+ "enable_tag_shuffling": false,
714
+ "tag_delimiter": ",",
715
+ "keep_tags_count": 1,
716
+ "tag_dropout_enable": false,
717
+ "tag_dropout_mode": "FULL",
718
+ "tag_dropout_probability": 0.0,
719
+ "tag_dropout_special_tags_mode": "NONE",
720
+ "tag_dropout_special_tags": "",
721
+ "tag_dropout_special_tags_regex": false,
722
+ "caps_randomize_enable": false,
723
+ "caps_randomize_mode": "capslock, title, first, random",
724
+ "caps_randomize_probability": 0.0,
725
+ "caps_randomize_lowercase": false
726
+ },
727
+ "name": "ladybonersgw-1536",
728
+ "path": "/workspace/Datasets/r_ladybonersgw/1792",
729
+ "seed": -157702382,
730
+ "enabled": true,
731
+ "type": "STANDARD",
732
+ "include_subdirectories": false,
733
+ "image_variations": 1,
734
+ "text_variations": 1,
735
+ "balancing": 1.0,
736
+ "balancing_strategy": "REPEATS",
737
+ "loss_weight": 1.0,
738
+ "concept_stats": {}
739
+ },
740
+ {
741
+ "__version": 2,
742
+ "image": {
743
+ "__version": 0,
744
+ "enable_crop_jitter": true,
745
+ "enable_random_flip": false,
746
+ "enable_fixed_flip": false,
747
+ "enable_random_rotate": false,
748
+ "enable_fixed_rotate": false,
749
+ "random_rotate_max_angle": 0.0,
750
+ "enable_random_brightness": false,
751
+ "enable_fixed_brightness": false,
752
+ "random_brightness_max_strength": 0.0,
753
+ "enable_random_contrast": false,
754
+ "enable_fixed_contrast": false,
755
+ "random_contrast_max_strength": 0.0,
756
+ "enable_random_saturation": false,
757
+ "enable_fixed_saturation": false,
758
+ "random_saturation_max_strength": 0.0,
759
+ "enable_random_hue": false,
760
+ "enable_fixed_hue": false,
761
+ "random_hue_max_strength": 0.0,
762
+ "enable_resolution_override": false,
763
+ "resolution_override": "512",
764
+ "enable_random_circular_mask_shrink": false,
765
+ "enable_random_mask_rotate_crop": false
766
+ },
767
+ "text": {
768
+ "__version": 0,
769
+ "prompt_source": "sample",
770
+ "prompt_path": "",
771
+ "enable_tag_shuffling": false,
772
+ "tag_delimiter": ",",
773
+ "keep_tags_count": 1,
774
+ "tag_dropout_enable": false,
775
+ "tag_dropout_mode": "FULL",
776
+ "tag_dropout_probability": 0.0,
777
+ "tag_dropout_special_tags_mode": "NONE",
778
+ "tag_dropout_special_tags": "",
779
+ "tag_dropout_special_tags_regex": false,
780
+ "caps_randomize_enable": false,
781
+ "caps_randomize_mode": "capslock, title, first, random",
782
+ "caps_randomize_probability": 0.0,
783
+ "caps_randomize_lowercase": false
784
+ },
785
+ "name": "ladybonersgw-2048",
786
+ "path": "/workspace/Datasets/r_ladybonersgw/2048",
787
+ "seed": -876890768,
788
+ "enabled": true,
789
+ "type": "STANDARD",
790
+ "include_subdirectories": false,
791
+ "image_variations": 1,
792
+ "text_variations": 1,
793
+ "balancing": 1.0,
794
+ "balancing_strategy": "REPEATS",
795
+ "loss_weight": 1.0,
796
+ "concept_stats": {}
797
+ },
798
+ {
799
+ "__version": 2,
800
+ "image": {
801
+ "__version": 0,
802
+ "enable_crop_jitter": true,
803
+ "enable_random_flip": false,
804
+ "enable_fixed_flip": false,
805
+ "enable_random_rotate": false,
806
+ "enable_fixed_rotate": false,
807
+ "random_rotate_max_angle": 0.0,
808
+ "enable_random_brightness": false,
809
+ "enable_fixed_brightness": false,
810
+ "random_brightness_max_strength": 0.0,
811
+ "enable_random_contrast": false,
812
+ "enable_fixed_contrast": false,
813
+ "random_contrast_max_strength": 0.0,
814
+ "enable_random_saturation": false,
815
+ "enable_fixed_saturation": false,
816
+ "random_saturation_max_strength": 0.0,
817
+ "enable_random_hue": false,
818
+ "enable_fixed_hue": false,
819
+ "random_hue_max_strength": 0.0,
820
+ "enable_resolution_override": true,
821
+ "resolution_override": "1024",
822
+ "enable_random_circular_mask_shrink": false,
823
+ "enable_random_mask_rotate_crop": false
824
+ },
825
+ "text": {
826
+ "__version": 0,
827
+ "prompt_source": "sample",
828
+ "prompt_path": "",
829
+ "enable_tag_shuffling": false,
830
+ "tag_delimiter": ",",
831
+ "keep_tags_count": 1,
832
+ "tag_dropout_enable": false,
833
+ "tag_dropout_mode": "FULL",
834
+ "tag_dropout_probability": 0.0,
835
+ "tag_dropout_special_tags_mode": "NONE",
836
+ "tag_dropout_special_tags": "",
837
+ "tag_dropout_special_tags_regex": false,
838
+ "caps_randomize_enable": false,
839
+ "caps_randomize_mode": "capslock, title, first, random",
840
+ "caps_randomize_probability": 0.0,
841
+ "caps_randomize_lowercase": false
842
+ },
843
+ "name": "ratemycock-1024",
844
+ "path": "/workspace/Datasets/r_ratemycock/1408",
845
+ "seed": 159728801,
846
+ "enabled": true,
847
+ "type": "STANDARD",
848
+ "include_subdirectories": false,
849
+ "image_variations": 1,
850
+ "text_variations": 1,
851
+ "balancing": 1.0,
852
+ "balancing_strategy": "REPEATS",
853
+ "loss_weight": 1.0,
854
+ "concept_stats": {
855
+ "file_size": 0,
856
+ "image_count": 0,
857
+ "image_with_mask_count": 0,
858
+ "image_with_caption_count": 0,
859
+ "video_count": 0,
860
+ "video_with_mask_count": 0,
861
+ "video_with_caption_count": 0,
862
+ "mask_count": 0,
863
+ "paired_masks": 0,
864
+ "unpaired_masks": 0,
865
+ "caption_count": 0,
866
+ "subcaption_count": 0,
867
+ "paired_captions": 0,
868
+ "unpaired_captions": 0,
869
+ "processing_time": 0.04560589790344238,
870
+ "directory_count": 1,
871
+ "max_pixels": [
872
+ 0,
873
+ "-",
874
+ "-"
875
+ ],
876
+ "min_pixels": [
877
+ 1000000000,
878
+ "-",
879
+ "-"
880
+ ],
881
+ "avg_pixels": 0,
882
+ "max_length": [
883
+ 0,
884
+ "-"
885
+ ],
886
+ "min_length": [
887
+ 1000000000,
888
+ "-"
889
+ ],
890
+ "avg_length": 0,
891
+ "max_fps": [
892
+ 0,
893
+ "-"
894
+ ],
895
+ "min_fps": [
896
+ 1000000000,
897
+ "-"
898
+ ],
899
+ "avg_fps": 0,
900
+ "max_caption_length": [
901
+ 0,
902
+ "-",
903
+ 0
904
+ ],
905
+ "min_caption_length": [
906
+ 1000000000,
907
+ "-",
908
+ 0
909
+ ],
910
+ "avg_caption_length": [
911
+ 0,
912
+ 0
913
+ ],
914
+ "aspect_buckets": {
915
+ "0.25": 0,
916
+ "0.285714": 0,
917
+ "0.333333": 0,
918
+ "0.4": 0,
919
+ "0.5": 0,
920
+ "0.571429": 0,
921
+ "0.666667": 0,
922
+ "0.8": 0,
923
+ "1.0": 0,
924
+ "1.25": 0,
925
+ "1.5": 0,
926
+ "1.75": 0,
927
+ "2.0": 0,
928
+ "2.5": 0,
929
+ "3.0": 0,
930
+ "3.5": 0,
931
+ "4.0": 0
932
+ },
933
+ "force_cancelled": false
934
+ }
935
+ },
936
+ {
937
+ "__version": 2,
938
+ "image": {
939
+ "__version": 0,
940
+ "enable_crop_jitter": true,
941
+ "enable_random_flip": false,
942
+ "enable_fixed_flip": false,
943
+ "enable_random_rotate": false,
944
+ "enable_fixed_rotate": false,
945
+ "random_rotate_max_angle": 0.0,
946
+ "enable_random_brightness": false,
947
+ "enable_fixed_brightness": false,
948
+ "random_brightness_max_strength": 0.0,
949
+ "enable_random_contrast": false,
950
+ "enable_fixed_contrast": false,
951
+ "random_contrast_max_strength": 0.0,
952
+ "enable_random_saturation": false,
953
+ "enable_fixed_saturation": false,
954
+ "random_saturation_max_strength": 0.0,
955
+ "enable_random_hue": false,
956
+ "enable_fixed_hue": false,
957
+ "random_hue_max_strength": 0.0,
958
+ "enable_resolution_override": false,
959
+ "resolution_override": "512",
960
+ "enable_random_circular_mask_shrink": false,
961
+ "enable_random_mask_rotate_crop": false
962
+ },
963
+ "text": {
964
+ "__version": 0,
965
+ "prompt_source": "sample",
966
+ "prompt_path": "",
967
+ "enable_tag_shuffling": false,
968
+ "tag_delimiter": ",",
969
+ "keep_tags_count": 1,
970
+ "tag_dropout_enable": false,
971
+ "tag_dropout_mode": "FULL",
972
+ "tag_dropout_probability": 0.0,
973
+ "tag_dropout_special_tags_mode": "NONE",
974
+ "tag_dropout_special_tags": "",
975
+ "tag_dropout_special_tags_regex": false,
976
+ "caps_randomize_enable": false,
977
+ "caps_randomize_mode": "capslock, title, first, random",
978
+ "caps_randomize_probability": 0.0,
979
+ "caps_randomize_lowercase": false
980
+ },
981
+ "name": "ratemycock-1536",
982
+ "path": "/workspace/Datasets/r_ratemycock/1792",
983
+ "seed": -249866332,
984
+ "enabled": true,
985
+ "type": "STANDARD",
986
+ "include_subdirectories": false,
987
+ "image_variations": 1,
988
+ "text_variations": 1,
989
+ "balancing": 1.0,
990
+ "balancing_strategy": "REPEATS",
991
+ "loss_weight": 1.0,
992
+ "concept_stats": {}
993
+ },
994
+ {
995
+ "__version": 2,
996
+ "image": {
997
+ "__version": 0,
998
+ "enable_crop_jitter": true,
999
+ "enable_random_flip": false,
1000
+ "enable_fixed_flip": false,
1001
+ "enable_random_rotate": false,
1002
+ "enable_fixed_rotate": false,
1003
+ "random_rotate_max_angle": 0.0,
1004
+ "enable_random_brightness": false,
1005
+ "enable_fixed_brightness": false,
1006
+ "random_brightness_max_strength": 0.0,
1007
+ "enable_random_contrast": false,
1008
+ "enable_fixed_contrast": false,
1009
+ "random_contrast_max_strength": 0.0,
1010
+ "enable_random_saturation": false,
1011
+ "enable_fixed_saturation": false,
1012
+ "random_saturation_max_strength": 0.0,
1013
+ "enable_random_hue": false,
1014
+ "enable_fixed_hue": false,
1015
+ "random_hue_max_strength": 0.0,
1016
+ "enable_resolution_override": false,
1017
+ "resolution_override": "512",
1018
+ "enable_random_circular_mask_shrink": false,
1019
+ "enable_random_mask_rotate_crop": false
1020
+ },
1021
+ "text": {
1022
+ "__version": 0,
1023
+ "prompt_source": "sample",
1024
+ "prompt_path": "",
1025
+ "enable_tag_shuffling": false,
1026
+ "tag_delimiter": ",",
1027
+ "keep_tags_count": 1,
1028
+ "tag_dropout_enable": false,
1029
+ "tag_dropout_mode": "FULL",
1030
+ "tag_dropout_probability": 0.0,
1031
+ "tag_dropout_special_tags_mode": "NONE",
1032
+ "tag_dropout_special_tags": "",
1033
+ "tag_dropout_special_tags_regex": false,
1034
+ "caps_randomize_enable": false,
1035
+ "caps_randomize_mode": "capslock, title, first, random",
1036
+ "caps_randomize_probability": 0.0,
1037
+ "caps_randomize_lowercase": false
1038
+ },
1039
+ "name": "ratemycock-2048",
1040
+ "path": "/workspace/Datasets/r_ratemycock/2048",
1041
+ "seed": -863006310,
1042
+ "enabled": true,
1043
+ "type": "STANDARD",
1044
+ "include_subdirectories": false,
1045
+ "image_variations": 1,
1046
+ "text_variations": 1,
1047
+ "balancing": 1.0,
1048
+ "balancing_strategy": "REPEATS",
1049
+ "loss_weight": 1.0,
1050
+ "concept_stats": {}
1051
+ }
1052
+ ],
1053
+ "aspect_ratio_bucketing": true,
1054
+ "latent_caching": true,
1055
+ "clear_cache_before_training": false,
1056
+ "learning_rate_scheduler": "COSINE",
1057
+ "custom_learning_rate_scheduler": null,
1058
+ "scheduler_params": [],
1059
+ "learning_rate": 0.0002,
1060
+ "learning_rate_warmup_steps": 50.0,
1061
+ "learning_rate_cycles": 1.0,
1062
+ "learning_rate_min_factor": 0.0,
1063
+ "epochs": 20,
1064
+ "batch_size": 1,
1065
+ "gradient_accumulation_steps": 2,
1066
+ "ema": "OFF",
1067
+ "ema_decay": 0.999,
1068
+ "ema_update_step_interval": 5,
1069
+ "dataloader_threads": 1,
1070
+ "train_device": "cuda",
1071
+ "temp_device": "cpu",
1072
+ "train_dtype": "BFLOAT_16",
1073
+ "fallback_train_dtype": "BFLOAT_16",
1074
+ "enable_autocast_cache": true,
1075
+ "only_cache": false,
1076
+ "resolution": "1024,1536",
1077
+ "frames": "25",
1078
+ "mse_strength": 1.0,
1079
+ "mae_strength": 0.0,
1080
+ "log_cosh_strength": 0.0,
1081
+ "huber_strength": 0.0,
1082
+ "huber_delta": 1.0,
1083
+ "vb_loss_strength": 1.0,
1084
+ "loss_weight_fn": "CONSTANT",
1085
+ "loss_weight_strength": 5.0,
1086
+ "dropout_probability": 0.0,
1087
+ "loss_scaler": "NONE",
1088
+ "learning_rate_scaler": "NONE",
1089
+ "clip_grad_norm": 1.0,
1090
+ "offset_noise_weight": 0.0,
1091
+ "generalized_offset_noise": false,
1092
+ "perturbation_noise_weight": 0.03,
1093
+ "rescale_noise_scheduler_to_zero_terminal_snr": false,
1094
+ "force_v_prediction": false,
1095
+ "force_epsilon_prediction": false,
1096
+ "min_noising_strength": 0.0,
1097
+ "max_noising_strength": 1.0,
1098
+ "timestep_distribution": "LOGIT_NORMAL",
1099
+ "noising_weight": 0.0,
1100
+ "noising_bias": 0.0,
1101
+ "timestep_shift": 3.0,
1102
+ "dynamic_timestep_shifting": false,
1103
+ "unet": {
1104
+ "__version": 0,
1105
+ "model_name": "",
1106
+ "include": true,
1107
+ "train": true,
1108
+ "stop_training_after": 0,
1109
+ "stop_training_after_unit": "NEVER",
1110
+ "learning_rate": null,
1111
+ "weight_dtype": "FLOAT_32",
1112
+ "dropout_probability": 0.0,
1113
+ "train_embedding": true,
1114
+ "attention_mask": false,
1115
+ "guidance_scale": 1.0
1116
+ },
1117
+ "prior": {
1118
+ "__version": 0,
1119
+ "model_name": "",
1120
+ "include": true,
1121
+ "train": true,
1122
+ "stop_training_after": 0,
1123
+ "stop_training_after_unit": "NEVER",
1124
+ "learning_rate": null,
1125
+ "weight_dtype": "FLOAT_32",
1126
+ "dropout_probability": 0.0,
1127
+ "train_embedding": true,
1128
+ "attention_mask": false,
1129
+ "guidance_scale": 1.0
1130
+ },
1131
+ "transformer": {
1132
+ "__version": 0,
1133
+ "model_name": "",
1134
+ "include": true,
1135
+ "train": true,
1136
+ "stop_training_after": 0,
1137
+ "stop_training_after_unit": "NEVER",
1138
+ "learning_rate": null,
1139
+ "weight_dtype": "FLOAT_8",
1140
+ "dropout_probability": 0.0,
1141
+ "train_embedding": true,
1142
+ "attention_mask": false,
1143
+ "guidance_scale": 1.0
1144
+ },
1145
+ "quantization": {
1146
+ "__version": 0,
1147
+ "layer_filter": "layers",
1148
+ "layer_filter_preset": "blocks",
1149
+ "layer_filter_regex": false,
1150
+ "svd_dtype": "NONE",
1151
+ "svd_rank": 16,
1152
+ "cache_dir": null
1153
+ },
1154
+ "text_encoder": {
1155
+ "__version": 0,
1156
+ "model_name": "",
1157
+ "include": true,
1158
+ "train": false,
1159
+ "stop_training_after": 30,
1160
+ "stop_training_after_unit": "EPOCH",
1161
+ "learning_rate": null,
1162
+ "weight_dtype": "FLOAT_8",
1163
+ "dropout_probability": 0.0,
1164
+ "train_embedding": true,
1165
+ "attention_mask": false,
1166
+ "guidance_scale": 1.0
1167
+ },
1168
+ "text_encoder_layer_skip": 0,
1169
+ "text_encoder_sequence_length": 512,
1170
+ "text_encoder_2": {
1171
+ "__version": 0,
1172
+ "model_name": "",
1173
+ "include": true,
1174
+ "train": true,
1175
+ "stop_training_after": 30,
1176
+ "stop_training_after_unit": "EPOCH",
1177
+ "learning_rate": null,
1178
+ "weight_dtype": "FLOAT_32",
1179
+ "dropout_probability": 0.0,
1180
+ "train_embedding": true,
1181
+ "attention_mask": false,
1182
+ "guidance_scale": 1.0
1183
+ },
1184
+ "text_encoder_2_layer_skip": 0,
1185
+ "text_encoder_2_sequence_length": 77,
1186
+ "text_encoder_3": {
1187
+ "__version": 0,
1188
+ "model_name": "",
1189
+ "include": true,
1190
+ "train": true,
1191
+ "stop_training_after": 30,
1192
+ "stop_training_after_unit": "EPOCH",
1193
+ "learning_rate": null,
1194
+ "weight_dtype": "FLOAT_32",
1195
+ "dropout_probability": 0.0,
1196
+ "train_embedding": true,
1197
+ "attention_mask": false,
1198
+ "guidance_scale": 1.0
1199
+ },
1200
+ "text_encoder_3_layer_skip": 0,
1201
+ "text_encoder_4": {
1202
+ "__version": 0,
1203
+ "model_name": "",
1204
+ "include": true,
1205
+ "train": true,
1206
+ "stop_training_after": 30,
1207
+ "stop_training_after_unit": "EPOCH",
1208
+ "learning_rate": null,
1209
+ "weight_dtype": "FLOAT_32",
1210
+ "dropout_probability": 0.0,
1211
+ "train_embedding": true,
1212
+ "attention_mask": false,
1213
+ "guidance_scale": 1.0
1214
+ },
1215
+ "text_encoder_4_layer_skip": 0,
1216
+ "vae": {
1217
+ "__version": 0,
1218
+ "model_name": "",
1219
+ "include": true,
1220
+ "train": true,
1221
+ "stop_training_after": null,
1222
+ "stop_training_after_unit": "NEVER",
1223
+ "learning_rate": null,
1224
+ "weight_dtype": "BFLOAT_16",
1225
+ "dropout_probability": 0.0,
1226
+ "train_embedding": true,
1227
+ "attention_mask": false,
1228
+ "guidance_scale": 1.0
1229
+ },
1230
+ "effnet_encoder": {
1231
+ "__version": 0,
1232
+ "model_name": "",
1233
+ "include": true,
1234
+ "train": true,
1235
+ "stop_training_after": null,
1236
+ "stop_training_after_unit": "NEVER",
1237
+ "learning_rate": null,
1238
+ "weight_dtype": "FLOAT_32",
1239
+ "dropout_probability": 0.0,
1240
+ "train_embedding": true,
1241
+ "attention_mask": false,
1242
+ "guidance_scale": 1.0
1243
+ },
1244
+ "decoder": {
1245
+ "__version": 0,
1246
+ "model_name": "",
1247
+ "include": true,
1248
+ "train": true,
1249
+ "stop_training_after": null,
1250
+ "stop_training_after_unit": "NEVER",
1251
+ "learning_rate": null,
1252
+ "weight_dtype": "FLOAT_32",
1253
+ "dropout_probability": 0.0,
1254
+ "train_embedding": true,
1255
+ "attention_mask": false,
1256
+ "guidance_scale": 1.0
1257
+ },
1258
+ "decoder_text_encoder": {
1259
+ "__version": 0,
1260
+ "model_name": "",
1261
+ "include": true,
1262
+ "train": true,
1263
+ "stop_training_after": null,
1264
+ "stop_training_after_unit": "NEVER",
1265
+ "learning_rate": null,
1266
+ "weight_dtype": "FLOAT_32",
1267
+ "dropout_probability": 0.0,
1268
+ "train_embedding": true,
1269
+ "attention_mask": false,
1270
+ "guidance_scale": 1.0
1271
+ },
1272
+ "decoder_vqgan": {
1273
+ "__version": 0,
1274
+ "model_name": "",
1275
+ "include": true,
1276
+ "train": true,
1277
+ "stop_training_after": null,
1278
+ "stop_training_after_unit": "NEVER",
1279
+ "learning_rate": null,
1280
+ "weight_dtype": "FLOAT_32",
1281
+ "dropout_probability": 0.0,
1282
+ "train_embedding": true,
1283
+ "attention_mask": false,
1284
+ "guidance_scale": 1.0
1285
+ },
1286
+ "masked_training": false,
1287
+ "unmasked_probability": 0.1,
1288
+ "unmasked_weight": 0.1,
1289
+ "normalize_masked_area_loss": false,
1290
+ "masked_prior_preservation_weight": 0.0,
1291
+ "custom_conditioning_image": false,
1292
+ "layer_filter": "^(?=.*attention)(?!.*refiner).*,^(?=.*feed_forward)(?!.*refiner).*",
1293
+ "layer_filter_preset": "attn-mlp",
1294
+ "layer_filter_regex": true,
1295
+ "embedding_learning_rate": null,
1296
+ "preserve_embedding_norm": false,
1297
+ "embedding": {
1298
+ "__version": 0,
1299
+ "uuid": "fdeb6425-1327-4993-aa4e-0f3f096c596a",
1300
+ "model_name": "",
1301
+ "placeholder": "<embedding>",
1302
+ "train": true,
1303
+ "stop_training_after": null,
1304
+ "stop_training_after_unit": "NEVER",
1305
+ "token_count": 1,
1306
+ "initial_embedding_text": "*",
1307
+ "is_output_embedding": false
1308
+ },
1309
+ "additional_embeddings": [],
1310
+ "embedding_weight_dtype": "FLOAT_32",
1311
+ "cloud": {
1312
+ "__version": 0,
1313
+ "enabled": false,
1314
+ "type": "RUNPOD",
1315
+ "file_sync": "NATIVE_SCP",
1316
+ "create": true,
1317
+ "name": "OneTrainer",
1318
+ "tensorboard_tunnel": true,
1319
+ "sub_type": "",
1320
+ "gpu_type": "",
1321
+ "volume_size": 100,
1322
+ "min_download": 0,
1323
+ "remote_dir": "/workspace",
1324
+ "huggingface_cache_dir": "/workspace/huggingface_cache",
1325
+ "onetrainer_dir": "/workspace/OneTrainer",
1326
+ "install_cmd": "git clone https://github.com/Nerogar/OneTrainer",
1327
+ "install_onetrainer": true,
1328
+ "update_onetrainer": true,
1329
+ "detach_trainer": false,
1330
+ "run_id": "job1",
1331
+ "download_samples": true,
1332
+ "download_output_model": true,
1333
+ "download_saves": true,
1334
+ "download_backups": false,
1335
+ "download_tensorboard": false,
1336
+ "delete_workspace": false,
1337
+ "on_finish": "NONE",
1338
+ "on_error": "NONE",
1339
+ "on_detached_finish": "NONE",
1340
+ "on_detached_error": "NONE"
1341
+ },
1342
+ "peft_type": "LORA",
1343
+ "lora_model_name": "",
1344
+ "lora_rank": 32,
1345
+ "lora_alpha": 16.0,
1346
+ "lora_decompose": true,
1347
+ "lora_decompose_norm_epsilon": true,
1348
+ "lora_decompose_output_axis": false,
1349
+ "lora_weight_dtype": "BFLOAT_16",
1350
+ "bundle_additional_embeddings": true,
1351
+ "oft_block_size": 32,
1352
+ "oft_coft": false,
1353
+ "coft_eps": 0.0001,
1354
+ "oft_block_share": false,
1355
+ "optimizer": {
1356
+ "__version": 0,
1357
+ "optimizer": "ADAMW_8BIT",
1358
+ "adam_w_mode": false,
1359
+ "alpha": null,
1360
+ "amsgrad": false,
1361
+ "beta1": 0.9,
1362
+ "beta2": 0.999,
1363
+ "beta3": null,
1364
+ "bias_correction": false,
1365
+ "block_wise": true,
1366
+ "capturable": false,
1367
+ "centered": false,
1368
+ "clip_threshold": null,
1369
+ "d0": null,
1370
+ "d_coef": null,
1371
+ "dampening": null,
1372
+ "decay_rate": null,
1373
+ "decouple": false,
1374
+ "differentiable": false,
1375
+ "eps": 1e-08,
1376
+ "eps2": null,
1377
+ "foreach": false,
1378
+ "fsdp_in_use": false,
1379
+ "fused": false,
1380
+ "fused_back_pass": false,
1381
+ "growth_rate": null,
1382
+ "initial_accumulator_value": null,
1383
+ "initial_accumulator": null,
1384
+ "is_paged": false,
1385
+ "log_every": null,
1386
+ "lr_decay": null,
1387
+ "max_unorm": null,
1388
+ "maximize": false,
1389
+ "min_8bit_size": 4096,
1390
+ "quant_block_size": null,
1391
+ "momentum": null,
1392
+ "nesterov": false,
1393
+ "no_prox": false,
1394
+ "optim_bits": 32,
1395
+ "percentile_clipping": 100,
1396
+ "r": null,
1397
+ "relative_step": false,
1398
+ "safeguard_warmup": false,
1399
+ "scale_parameter": false,
1400
+ "stochastic_rounding": true,
1401
+ "use_bias_correction": false,
1402
+ "use_triton": false,
1403
+ "warmup_init": false,
1404
+ "weight_decay": 0.01,
1405
+ "weight_lr_power": null,
1406
+ "decoupled_decay": false,
1407
+ "fixed_decay": false,
1408
+ "rectify": false,
1409
+ "degenerated_to_sgd": false,
1410
+ "k": null,
1411
+ "xi": null,
1412
+ "n_sma_threshold": null,
1413
+ "ams_bound": false,
1414
+ "adanorm": false,
1415
+ "adam_debias": false,
1416
+ "slice_p": null,
1417
+ "cautious": false,
1418
+ "weight_decay_by_lr": true,
1419
+ "prodigy_steps": null,
1420
+ "use_speed": false,
1421
+ "split_groups": true,
1422
+ "split_groups_mean": true,
1423
+ "factored": true,
1424
+ "factored_fp32": true,
1425
+ "use_stableadamw": true,
1426
+ "use_cautious": false,
1427
+ "use_grams": false,
1428
+ "use_adopt": false,
1429
+ "d_limiter": true,
1430
+ "use_schedulefree": true,
1431
+ "use_orthograd": false,
1432
+ "nnmf_factor": false,
1433
+ "orthogonal_gradient": false,
1434
+ "use_atan2": false,
1435
+ "use_AdEMAMix": false,
1436
+ "beta3_ema": null,
1437
+ "alpha_grad": null,
1438
+ "beta1_warmup": null,
1439
+ "min_beta1": null,
1440
+ "Simplified_AdEMAMix": false,
1441
+ "cautious_mask": false,
1442
+ "grams_moment": false,
1443
+ "kourkoutas_beta": false,
1444
+ "k_warmup_steps": null,
1445
+ "schedulefree_c": null,
1446
+ "ns_steps": null,
1447
+ "MuonWithAuxAdam": false,
1448
+ "muon_hidden_layers": null,
1449
+ "muon_adam_regex": false,
1450
+ "muon_adam_lr": null,
1451
+ "muon_te1_adam_lr": null,
1452
+ "muon_te2_adam_lr": null,
1453
+ "muon_adam_config": null,
1454
+ "rms_rescaling": true,
1455
+ "normuon_variant": false,
1456
+ "beta2_normuon": null,
1457
+ "normuon_eps": null,
1458
+ "low_rank_ortho": false,
1459
+ "ortho_rank": null,
1460
+ "accelerated_ns": false,
1461
+ "cautious_wd": false,
1462
+ "approx_mars": false,
1463
+ "kappa_p": null,
1464
+ "auto_kappa_p": false,
1465
+ "compile": false
1466
+ },
1467
+ "optimizer_defaults": {},
1468
+ "sample_definition_file_name": "training_samples/samples.json",
1469
+ "samples": [
1470
+ {
1471
+ "__version": 0,
1472
+ "enabled": true,
1473
+ "prompt": "A high-angle selfie photograph \n captures a light-skinned man in his 30s lying in bed. He is propped up \n against several large, off-white pillows, looking directly at the camera with \n a gentle smile. The man has a bald head with some dark hair on the sides, a \n dark goatee, and a hairy chest and abdomen. His right arm is extended, \n holding the camera, while his left hand is positioned near his groin, \n grasping his semi-erect penis. The setting is intimate and casual, with the \n bed covered in a mix of blue and brown blankets.\\n\\nThe image is a realistic, \n unfiltered photograph with warm, indoor lighting that casts soft shadows \n across his body and the bedding. The dominant color palette is composed of \n neutral tones like cream, tan, and blue, which contrast with his natural skin \n tones and dark body hair. The composition is a medium close-up, focusing on \n his upper body and face, creating a sense of personal connection and \n vulnerability.\\n\\nThe man's physical features are prominent, including thick \n chest hair that tapers down his belly and dense dark hair in his armpit. His \n semi-erect penis is clearly visible, held by his hand. The overall mood of \n the image is relaxed and comfortable, conveying a sense of self-assuredness \n and intimacy. The man's expression is one of contentment, and the informal \n nature of the shot suggests a private moment captured in a personal \n setting.",
1474
+ "negative_prompt": "",
1475
+ "height": 800,
1476
+ "width": 1200,
1477
+ "frames": 1,
1478
+ "length": 10.0,
1479
+ "seed": 1142,
1480
+ "random_seed": false,
1481
+ "diffusion_steps": 20,
1482
+ "cfg_scale": 7.0,
1483
+ "noise_scheduler": "DDIM",
1484
+ "text_encoder_1_layer_skip": 0,
1485
+ "text_encoder_1_sequence_length": null,
1486
+ "text_encoder_2_layer_skip": 0,
1487
+ "text_encoder_2_sequence_length": null,
1488
+ "text_encoder_3_layer_skip": 0,
1489
+ "text_encoder_4_layer_skip": 0,
1490
+ "transformer_attention_mask": false,
1491
+ "force_last_timestep": false,
1492
+ "sample_inpainting": false,
1493
+ "base_image_path": "",
1494
+ "mask_image_path": ""
1495
+ },
1496
+ {
1497
+ "__version": 0,
1498
+ "enabled": true,
1499
+ "prompt": "A full-frontal photograph features \n a man in his 30s with a muscular, hairy physique and a prominent dark \n mustache and goatee. He is seated with his legs spread wide, facing the \n camera directly. His arms are raised, and his hands are tucked behind his \n head, adjusting or holding a camouflage-patterned baseball cap. He has a \n neutral, intense gaze directed toward the viewer. The setting appears to be \n an indoor room, with a grey-patterned curtain visible on the left and a \n framed art piece depicting a deer in a snowy woods on the right wall.\\n\\nThe \n image is a realistic photograph with an amateur, intimate quality. The \n lighting is interior and slightly dim, originating from above and in front of \n the subject, which creates soft shadows under his arms and highlights the \n contours of his torso. The color palette is dominated by natural flesh tones, \n dark body hair, and the neutral grey and brown tones of the background \n elements.\\n\\nThe man is completely nude, displaying full frontal exposure. \n His torso is covered in thick, dark hair that extends from his chest down to \n his stomach and into a dense pubic region. His genitals are centrally framed \n and fully visible; his penis is in a flaccid or semi-erect state, resting \n over his scrotum. His legs are spread wide, showing hairy thighs and more of \n the groin area. The low-angle composition emphasizes his body and genitals, \n conveying a sense of exhibitionism and confidence. The combination of his \n pose, facial expression, and the rustic-themed decor in the background \n suggests a deliberate presentation of raw, masculine energy.",
1500
+ "negative_prompt": "",
1501
+ "height": 1536,
1502
+ "width": 1536,
1503
+ "frames": 1,
1504
+ "length": 10.0,
1505
+ "seed": 2242,
1506
+ "random_seed": false,
1507
+ "diffusion_steps": 20,
1508
+ "cfg_scale": 7.0,
1509
+ "noise_scheduler": "DDIM",
1510
+ "text_encoder_1_layer_skip": 0,
1511
+ "text_encoder_1_sequence_length": null,
1512
+ "text_encoder_2_layer_skip": 0,
1513
+ "text_encoder_2_sequence_length": null,
1514
+ "text_encoder_3_layer_skip": 0,
1515
+ "text_encoder_4_layer_skip": 0,
1516
+ "transformer_attention_mask": false,
1517
+ "force_last_timestep": false,
1518
+ "sample_inpainting": false,
1519
+ "base_image_path": "",
1520
+ "mask_image_path": ""
1521
+ },
1522
+ {
1523
+ "__version": 0,
1524
+ "enabled": true,
1525
+ "prompt": "This realistic photograph features \n a fit, light-skinned adult man in his 30s with short, sandy-blond hair and a \n well-defined, athletic physique. He is standing in a bathroom, taking a \n mirror selfie. He holds a white smartphone in his right hand while using his \n left hand to pull down the waistband of his blue Jockey boxer briefs, \n exposing his lower torso and groin area. The background shows a portion of a \n bedroom with a neatly made bed and a gray textured shower curtain to his \n right.\\n\\nThe image is a medium-framed shot captured with clear, even indoor \n lighting that emphasizes the contours of the man's chest and abdominal \n muscles. The color palette is composed of neutral tones, including the beige \n bathroom walls, the off-white bedroom beyond the doorway, and the man's fair \n skin. The overall mood is casual and intimate, typical of a personal \n self-portrait.\\n\\nIn a state of partial nudity, the man has pulled his \n underwear down to reveal his pubic region. He has a patch of reddish-brown \n pubic hair. His flaccid penis is fully visible, resting over the lowered \n waistband of his underwear. The exposure is deliberate and presented in a \n straightforward manner.\\n\\nThe man maintains a neutral, somewhat focused \n expression as he looks toward the mirror to frame his shot. The setting is a \n standard domestic environment, and the inclusion of the bedroom in the \n background enhances the private, personal nature of the photograph. The image \n captures a moment of physical self-documentation in a private space.",
1526
+ "negative_prompt": "",
1527
+ "height": 2400,
1528
+ "width": 1600,
1529
+ "frames": 1,
1530
+ "length": 10.0,
1531
+ "seed": 3342,
1532
+ "random_seed": false,
1533
+ "diffusion_steps": 20,
1534
+ "cfg_scale": 7.0,
1535
+ "noise_scheduler": "DDIM",
1536
+ "text_encoder_1_layer_skip": 0,
1537
+ "text_encoder_1_sequence_length": null,
1538
+ "text_encoder_2_layer_skip": 0,
1539
+ "text_encoder_2_sequence_length": null,
1540
+ "text_encoder_3_layer_skip": 0,
1541
+ "text_encoder_4_layer_skip": 0,
1542
+ "transformer_attention_mask": false,
1543
+ "force_last_timestep": false,
1544
+ "sample_inpainting": false,
1545
+ "base_image_path": "",
1546
+ "mask_image_path": ""
1547
+ }
1548
+ ],
1549
+ "sample_after": 200,
1550
+ "sample_after_unit": "STEP",
1551
+ "sample_skip_first": 0,
1552
+ "sample_image_format": "JPG",
1553
+ "sample_video_format": "MP4",
1554
+ "sample_audio_format": "MP3",
1555
+ "samples_to_tensorboard": true,
1556
+ "non_ema_sampling": true,
1557
+ "backup_after": 30,
1558
+ "backup_after_unit": "MINUTE",
1559
+ "rolling_backup": false,
1560
+ "rolling_backup_count": 3,
1561
+ "backup_before_save": true,
1562
+ "save_every": 200,
1563
+ "save_every_unit": "STEP",
1564
+ "save_skip_first": 0,
1565
+ "save_filename_prefix": "reddit-2-zib"
1566
+ }
ZIB/config/reddit-2-zib2026-02-15_23-12-50.json ADDED
@@ -0,0 +1,1566 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version": 10,
3
+ "training_method": "LORA",
4
+ "model_type": "Z_IMAGE",
5
+ "debug_mode": false,
6
+ "debug_dir": "debug",
7
+ "workspace_dir": "workspace/run",
8
+ "cache_dir": "workspace-cache/run",
9
+ "tensorboard": true,
10
+ "tensorboard_expose": false,
11
+ "tensorboard_always_on": false,
12
+ "tensorboard_port": 6007,
13
+ "validation": false,
14
+ "validate_after": 1,
15
+ "validate_after_unit": "EPOCH",
16
+ "continue_last_backup": false,
17
+ "include_train_config": "NONE",
18
+ "multi_gpu": true,
19
+ "device_indexes": "0,1",
20
+ "gradient_reduce_precision": "FLOAT_32_STOCHASTIC",
21
+ "fused_gradient_reduce": true,
22
+ "async_gradient_reduce": true,
23
+ "async_gradient_reduce_buffer": 100,
24
+ "base_model_name": "Tongyi-MAI/Z-Image",
25
+ "output_dtype": "BFLOAT_16",
26
+ "output_model_format": "SAFETENSORS",
27
+ "output_model_destination": "models/reddit-2-zib.safetensors",
28
+ "gradient_checkpointing": "ON",
29
+ "enable_async_offloading": true,
30
+ "enable_activation_offloading": true,
31
+ "layer_offload_fraction": 0.001,
32
+ "force_circular_padding": false,
33
+ "compile": true,
34
+ "concept_file_name": "training_concepts/concepts.json",
35
+ "concepts": [
36
+ {
37
+ "__version": 2,
38
+ "image": {
39
+ "__version": 0,
40
+ "enable_crop_jitter": true,
41
+ "enable_random_flip": false,
42
+ "enable_fixed_flip": false,
43
+ "enable_random_rotate": false,
44
+ "enable_fixed_rotate": false,
45
+ "random_rotate_max_angle": 0.0,
46
+ "enable_random_brightness": false,
47
+ "enable_fixed_brightness": false,
48
+ "random_brightness_max_strength": 0.0,
49
+ "enable_random_contrast": false,
50
+ "enable_fixed_contrast": false,
51
+ "random_contrast_max_strength": 0.0,
52
+ "enable_random_saturation": false,
53
+ "enable_fixed_saturation": false,
54
+ "random_saturation_max_strength": 0.0,
55
+ "enable_random_hue": false,
56
+ "enable_fixed_hue": false,
57
+ "random_hue_max_strength": 0.0,
58
+ "enable_resolution_override": true,
59
+ "resolution_override": "1024",
60
+ "enable_random_circular_mask_shrink": false,
61
+ "enable_random_mask_rotate_crop": false
62
+ },
63
+ "text": {
64
+ "__version": 0,
65
+ "prompt_source": "sample",
66
+ "prompt_path": "",
67
+ "enable_tag_shuffling": false,
68
+ "tag_delimiter": ",",
69
+ "keep_tags_count": 1,
70
+ "tag_dropout_enable": false,
71
+ "tag_dropout_mode": "FULL",
72
+ "tag_dropout_probability": 0.0,
73
+ "tag_dropout_special_tags_mode": "NONE",
74
+ "tag_dropout_special_tags": "",
75
+ "tag_dropout_special_tags_regex": false,
76
+ "caps_randomize_enable": false,
77
+ "caps_randomize_mode": "capslock, title, first, random",
78
+ "caps_randomize_probability": 0.0,
79
+ "caps_randomize_lowercase": false
80
+ },
81
+ "name": "cock-1024",
82
+ "path": "/workspace/Datasets/r_cock/1408",
83
+ "seed": -151055228,
84
+ "enabled": true,
85
+ "type": "STANDARD",
86
+ "include_subdirectories": false,
87
+ "image_variations": 1,
88
+ "text_variations": 1,
89
+ "balancing": 1.0,
90
+ "balancing_strategy": "REPEATS",
91
+ "loss_weight": 1.0,
92
+ "concept_stats": {
93
+ "file_size": 18979799,
94
+ "image_count": 84,
95
+ "image_with_mask_count": 0,
96
+ "image_with_caption_count": 84,
97
+ "video_count": 0,
98
+ "video_with_mask_count": 0,
99
+ "video_with_caption_count": 0,
100
+ "mask_count": 0,
101
+ "paired_masks": 0,
102
+ "unpaired_masks": 0,
103
+ "caption_count": 84,
104
+ "subcaption_count": 468,
105
+ "paired_captions": 84,
106
+ "unpaired_captions": 0,
107
+ "processing_time": 0.022084553027525544,
108
+ "directory_count": 1,
109
+ "max_pixels": [
110
+ 1972080,
111
+ "cock_2022-12-11_03-13-22.jpg",
112
+ "1080w x 1826h"
113
+ ],
114
+ "min_pixels": [
115
+ 1065636,
116
+ "cock_2026-01-10_08-01-31_03.webp",
117
+ "1242w x 858h"
118
+ ],
119
+ "avg_pixels": 1498911.7261904762,
120
+ "max_length": [
121
+ 0,
122
+ "-"
123
+ ],
124
+ "min_length": [
125
+ 1000000000,
126
+ "-"
127
+ ],
128
+ "avg_length": 0,
129
+ "max_fps": [
130
+ 0,
131
+ "-"
132
+ ],
133
+ "min_fps": [
134
+ 1000000000,
135
+ "-"
136
+ ],
137
+ "avg_fps": 0,
138
+ "max_caption_length": [
139
+ 892,
140
+ "cock_2026-02-13_06-20-19.jpg",
141
+ 136
142
+ ],
143
+ "min_caption_length": [
144
+ 0,
145
+ "cock_2019-07-24_12-02-42.jpg",
146
+ 0
147
+ ],
148
+ "avg_caption_length": [
149
+ 306.34670873400086,
150
+ 50.15109395168455
151
+ ],
152
+ "aspect_buckets": {
153
+ "0.25": 0,
154
+ "0.285714": 0,
155
+ "0.333333": 0,
156
+ "0.4": 0,
157
+ "0.5": 3,
158
+ "0.571429": 0,
159
+ "0.666667": 1,
160
+ "0.8": 2,
161
+ "1.0": 4,
162
+ "1.25": 43,
163
+ "1.5": 9,
164
+ "1.75": 17,
165
+ "2.0": 5,
166
+ "2.5": 0,
167
+ "3.0": 0,
168
+ "3.5": 0,
169
+ "4.0": 0
170
+ },
171
+ "force_cancelled": false
172
+ }
173
+ },
174
+ {
175
+ "__version": 2,
176
+ "image": {
177
+ "__version": 0,
178
+ "enable_crop_jitter": true,
179
+ "enable_random_flip": false,
180
+ "enable_fixed_flip": false,
181
+ "enable_random_rotate": false,
182
+ "enable_fixed_rotate": false,
183
+ "random_rotate_max_angle": 0.0,
184
+ "enable_random_brightness": false,
185
+ "enable_fixed_brightness": false,
186
+ "random_brightness_max_strength": 0.0,
187
+ "enable_random_contrast": false,
188
+ "enable_fixed_contrast": false,
189
+ "random_contrast_max_strength": 0.0,
190
+ "enable_random_saturation": false,
191
+ "enable_fixed_saturation": false,
192
+ "random_saturation_max_strength": 0.0,
193
+ "enable_random_hue": false,
194
+ "enable_fixed_hue": false,
195
+ "random_hue_max_strength": 0.0,
196
+ "enable_resolution_override": false,
197
+ "resolution_override": "1024,1536",
198
+ "enable_random_circular_mask_shrink": false,
199
+ "enable_random_mask_rotate_crop": false
200
+ },
201
+ "text": {
202
+ "__version": 0,
203
+ "prompt_source": "sample",
204
+ "prompt_path": "",
205
+ "enable_tag_shuffling": false,
206
+ "tag_delimiter": ",",
207
+ "keep_tags_count": 1,
208
+ "tag_dropout_enable": false,
209
+ "tag_dropout_mode": "FULL",
210
+ "tag_dropout_probability": 0.0,
211
+ "tag_dropout_special_tags_mode": "NONE",
212
+ "tag_dropout_special_tags": "",
213
+ "tag_dropout_special_tags_regex": false,
214
+ "caps_randomize_enable": false,
215
+ "caps_randomize_mode": "capslock, title, first, random",
216
+ "caps_randomize_probability": 0.0,
217
+ "caps_randomize_lowercase": false
218
+ },
219
+ "name": "cock-1536",
220
+ "path": "/workspace/Datasets/r_cock/1792",
221
+ "seed": 857416178,
222
+ "enabled": true,
223
+ "type": "STANDARD",
224
+ "include_subdirectories": false,
225
+ "image_variations": 1,
226
+ "text_variations": 1,
227
+ "balancing": 1.0,
228
+ "balancing_strategy": "REPEATS",
229
+ "loss_weight": 1.0,
230
+ "concept_stats": {
231
+ "file_size": 26015793,
232
+ "image_count": 61,
233
+ "image_with_mask_count": 0,
234
+ "image_with_caption_count": 61,
235
+ "video_count": 0,
236
+ "video_with_mask_count": 0,
237
+ "video_with_caption_count": 0,
238
+ "mask_count": 0,
239
+ "paired_masks": 0,
240
+ "unpaired_masks": 0,
241
+ "caption_count": 61,
242
+ "subcaption_count": 333,
243
+ "paired_captions": 61,
244
+ "unpaired_captions": 0,
245
+ "processing_time": 0.016968363896012306,
246
+ "directory_count": 1,
247
+ "max_pixels": [
248
+ 3165170,
249
+ "cock_2020-06-22_12-19-13.jpg",
250
+ "1535w x 2062h"
251
+ ],
252
+ "min_pixels": [
253
+ 1984940,
254
+ "cock_2026-02-13_18-10-26.jpg",
255
+ "1220w x 1627h"
256
+ ],
257
+ "avg_pixels": 2598588.032786885,
258
+ "max_length": [
259
+ 0,
260
+ "-"
261
+ ],
262
+ "min_length": [
263
+ 1000000000,
264
+ "-"
265
+ ],
266
+ "avg_length": 0,
267
+ "max_fps": [
268
+ 0,
269
+ "-"
270
+ ],
271
+ "min_fps": [
272
+ 1000000000,
273
+ "-"
274
+ ],
275
+ "avg_fps": 0,
276
+ "max_caption_length": [
277
+ 811,
278
+ "cock_2025-11-26_13-35-55_01.webp",
279
+ 135
280
+ ],
281
+ "min_caption_length": [
282
+ 0,
283
+ "cock_2019-02-05_08-32-21.jpg",
284
+ 0
285
+ ],
286
+ "avg_caption_length": [
287
+ 311.8868407765503,
288
+ 50.57262830257678
289
+ ],
290
+ "aspect_buckets": {
291
+ "0.25": 0,
292
+ "0.285714": 0,
293
+ "0.333333": 0,
294
+ "0.4": 0,
295
+ "0.5": 0,
296
+ "0.571429": 0,
297
+ "0.666667": 1,
298
+ "0.8": 3,
299
+ "1.0": 2,
300
+ "1.25": 16,
301
+ "1.5": 7,
302
+ "1.75": 27,
303
+ "2.0": 5,
304
+ "2.5": 0,
305
+ "3.0": 0,
306
+ "3.5": 0,
307
+ "4.0": 0
308
+ },
309
+ "force_cancelled": false
310
+ }
311
+ },
312
+ {
313
+ "__version": 2,
314
+ "image": {
315
+ "__version": 0,
316
+ "enable_crop_jitter": true,
317
+ "enable_random_flip": false,
318
+ "enable_fixed_flip": false,
319
+ "enable_random_rotate": false,
320
+ "enable_fixed_rotate": false,
321
+ "random_rotate_max_angle": 0.0,
322
+ "enable_random_brightness": false,
323
+ "enable_fixed_brightness": false,
324
+ "random_brightness_max_strength": 0.0,
325
+ "enable_random_contrast": false,
326
+ "enable_fixed_contrast": false,
327
+ "random_contrast_max_strength": 0.0,
328
+ "enable_random_saturation": false,
329
+ "enable_fixed_saturation": false,
330
+ "random_saturation_max_strength": 0.0,
331
+ "enable_random_hue": false,
332
+ "enable_fixed_hue": false,
333
+ "random_hue_max_strength": 0.0,
334
+ "enable_resolution_override": false,
335
+ "resolution_override": "512",
336
+ "enable_random_circular_mask_shrink": false,
337
+ "enable_random_mask_rotate_crop": false
338
+ },
339
+ "text": {
340
+ "__version": 0,
341
+ "prompt_source": "sample",
342
+ "prompt_path": "",
343
+ "enable_tag_shuffling": false,
344
+ "tag_delimiter": ",",
345
+ "keep_tags_count": 1,
346
+ "tag_dropout_enable": false,
347
+ "tag_dropout_mode": "FULL",
348
+ "tag_dropout_probability": 0.0,
349
+ "tag_dropout_special_tags_mode": "NONE",
350
+ "tag_dropout_special_tags": "",
351
+ "tag_dropout_special_tags_regex": false,
352
+ "caps_randomize_enable": false,
353
+ "caps_randomize_mode": "capslock, title, first, random",
354
+ "caps_randomize_probability": 0.0,
355
+ "caps_randomize_lowercase": false
356
+ },
357
+ "name": "cock-2048",
358
+ "path": "/workspace/Datasets/r_cock/2048",
359
+ "seed": -313616135,
360
+ "enabled": true,
361
+ "type": "STANDARD",
362
+ "include_subdirectories": false,
363
+ "image_variations": 1,
364
+ "text_variations": 1,
365
+ "balancing": 1.0,
366
+ "balancing_strategy": "REPEATS",
367
+ "loss_weight": 1.0,
368
+ "concept_stats": {
369
+ "file_size": 413667396,
370
+ "image_count": 351,
371
+ "image_with_mask_count": 0,
372
+ "image_with_caption_count": 351,
373
+ "video_count": 0,
374
+ "video_with_mask_count": 0,
375
+ "video_with_caption_count": 0,
376
+ "mask_count": 0,
377
+ "paired_masks": 0,
378
+ "unpaired_masks": 0,
379
+ "caption_count": 351,
380
+ "subcaption_count": 1941,
381
+ "paired_captions": 351,
382
+ "unpaired_captions": 0,
383
+ "processing_time": 0.05115061393007636,
384
+ "directory_count": 1,
385
+ "max_pixels": [
386
+ 49939200,
387
+ "cock_2026-02-12_19-39-44.jpg",
388
+ "6120w x 8160h"
389
+ ],
390
+ "min_pixels": [
391
+ 3232800,
392
+ "cock_2019-12-26_09-52-24.png",
393
+ "1440w x 2245h"
394
+ ],
395
+ "avg_pixels": 9450439.341880342,
396
+ "max_length": [
397
+ 0,
398
+ "-"
399
+ ],
400
+ "min_length": [
401
+ 1000000000,
402
+ "-"
403
+ ],
404
+ "avg_length": 0,
405
+ "max_fps": [
406
+ 0,
407
+ "-"
408
+ ],
409
+ "min_fps": [
410
+ 1000000000,
411
+ "-"
412
+ ],
413
+ "avg_fps": 0,
414
+ "max_caption_length": [
415
+ 913,
416
+ "cock_2025-02-23_16-15-28_03.webp",
417
+ 145
418
+ ],
419
+ "min_caption_length": [
420
+ 0,
421
+ "cock_2019-02-20_17-33-55.jpg",
422
+ 0
423
+ ],
424
+ "avg_caption_length": [
425
+ 308.8869282186318,
426
+ 50.642955884895436
427
+ ],
428
+ "aspect_buckets": {
429
+ "0.25": 0,
430
+ "0.285714": 0,
431
+ "0.333333": 0,
432
+ "0.4": 1,
433
+ "0.5": 0,
434
+ "0.571429": 6,
435
+ "0.666667": 3,
436
+ "0.8": 54,
437
+ "1.0": 17,
438
+ "1.25": 219,
439
+ "1.5": 23,
440
+ "1.75": 18,
441
+ "2.0": 9,
442
+ "2.5": 0,
443
+ "3.0": 1,
444
+ "3.5": 0,
445
+ "4.0": 0
446
+ },
447
+ "force_cancelled": false
448
+ }
449
+ },
450
+ {
451
+ "__version": 2,
452
+ "image": {
453
+ "__version": 0,
454
+ "enable_crop_jitter": true,
455
+ "enable_random_flip": false,
456
+ "enable_fixed_flip": false,
457
+ "enable_random_rotate": false,
458
+ "enable_fixed_rotate": false,
459
+ "random_rotate_max_angle": 0.0,
460
+ "enable_random_brightness": false,
461
+ "enable_fixed_brightness": false,
462
+ "random_brightness_max_strength": 0.0,
463
+ "enable_random_contrast": false,
464
+ "enable_fixed_contrast": false,
465
+ "random_contrast_max_strength": 0.0,
466
+ "enable_random_saturation": false,
467
+ "enable_fixed_saturation": false,
468
+ "random_saturation_max_strength": 0.0,
469
+ "enable_random_hue": false,
470
+ "enable_fixed_hue": false,
471
+ "random_hue_max_strength": 0.0,
472
+ "enable_resolution_override": true,
473
+ "resolution_override": "1024",
474
+ "enable_random_circular_mask_shrink": false,
475
+ "enable_random_mask_rotate_crop": false
476
+ },
477
+ "text": {
478
+ "__version": 0,
479
+ "prompt_source": "sample",
480
+ "prompt_path": "",
481
+ "enable_tag_shuffling": false,
482
+ "tag_delimiter": ",",
483
+ "keep_tags_count": 1,
484
+ "tag_dropout_enable": false,
485
+ "tag_dropout_mode": "FULL",
486
+ "tag_dropout_probability": 0.0,
487
+ "tag_dropout_special_tags_mode": "NONE",
488
+ "tag_dropout_special_tags": "",
489
+ "tag_dropout_special_tags_regex": false,
490
+ "caps_randomize_enable": false,
491
+ "caps_randomize_mode": "capslock, title, first, random",
492
+ "caps_randomize_probability": 0.0,
493
+ "caps_randomize_lowercase": false
494
+ },
495
+ "name": "gayfaceandcock-1024",
496
+ "path": "/workspace/Datasets/r_gayfaceandcock/1384",
497
+ "seed": -51838343,
498
+ "enabled": true,
499
+ "type": "STANDARD",
500
+ "include_subdirectories": false,
501
+ "image_variations": 1,
502
+ "text_variations": 1,
503
+ "balancing": 1.0,
504
+ "balancing_strategy": "REPEATS",
505
+ "loss_weight": 1.0,
506
+ "concept_stats": {}
507
+ },
508
+ {
509
+ "__version": 2,
510
+ "image": {
511
+ "__version": 0,
512
+ "enable_crop_jitter": true,
513
+ "enable_random_flip": false,
514
+ "enable_fixed_flip": false,
515
+ "enable_random_rotate": false,
516
+ "enable_fixed_rotate": false,
517
+ "random_rotate_max_angle": 0.0,
518
+ "enable_random_brightness": false,
519
+ "enable_fixed_brightness": false,
520
+ "random_brightness_max_strength": 0.0,
521
+ "enable_random_contrast": false,
522
+ "enable_fixed_contrast": false,
523
+ "random_contrast_max_strength": 0.0,
524
+ "enable_random_saturation": false,
525
+ "enable_fixed_saturation": false,
526
+ "random_saturation_max_strength": 0.0,
527
+ "enable_random_hue": false,
528
+ "enable_fixed_hue": false,
529
+ "random_hue_max_strength": 0.0,
530
+ "enable_resolution_override": false,
531
+ "resolution_override": "512",
532
+ "enable_random_circular_mask_shrink": false,
533
+ "enable_random_mask_rotate_crop": false
534
+ },
535
+ "text": {
536
+ "__version": 0,
537
+ "prompt_source": "sample",
538
+ "prompt_path": "",
539
+ "enable_tag_shuffling": false,
540
+ "tag_delimiter": ",",
541
+ "keep_tags_count": 1,
542
+ "tag_dropout_enable": false,
543
+ "tag_dropout_mode": "FULL",
544
+ "tag_dropout_probability": 0.0,
545
+ "tag_dropout_special_tags_mode": "NONE",
546
+ "tag_dropout_special_tags": "",
547
+ "tag_dropout_special_tags_regex": false,
548
+ "caps_randomize_enable": false,
549
+ "caps_randomize_mode": "capslock, title, first, random",
550
+ "caps_randomize_probability": 0.0,
551
+ "caps_randomize_lowercase": false
552
+ },
553
+ "name": "gayfaceandcock-1536",
554
+ "path": "/workspace/Datasets/r_gayfaceandcock/1792",
555
+ "seed": -461368761,
556
+ "enabled": true,
557
+ "type": "STANDARD",
558
+ "include_subdirectories": false,
559
+ "image_variations": 1,
560
+ "text_variations": 1,
561
+ "balancing": 1.0,
562
+ "balancing_strategy": "REPEATS",
563
+ "loss_weight": 1.0,
564
+ "concept_stats": {}
565
+ },
566
+ {
567
+ "__version": 2,
568
+ "image": {
569
+ "__version": 0,
570
+ "enable_crop_jitter": true,
571
+ "enable_random_flip": false,
572
+ "enable_fixed_flip": false,
573
+ "enable_random_rotate": false,
574
+ "enable_fixed_rotate": false,
575
+ "random_rotate_max_angle": 0.0,
576
+ "enable_random_brightness": false,
577
+ "enable_fixed_brightness": false,
578
+ "random_brightness_max_strength": 0.0,
579
+ "enable_random_contrast": false,
580
+ "enable_fixed_contrast": false,
581
+ "random_contrast_max_strength": 0.0,
582
+ "enable_random_saturation": false,
583
+ "enable_fixed_saturation": false,
584
+ "random_saturation_max_strength": 0.0,
585
+ "enable_random_hue": false,
586
+ "enable_fixed_hue": false,
587
+ "random_hue_max_strength": 0.0,
588
+ "enable_resolution_override": false,
589
+ "resolution_override": "512",
590
+ "enable_random_circular_mask_shrink": false,
591
+ "enable_random_mask_rotate_crop": false
592
+ },
593
+ "text": {
594
+ "__version": 0,
595
+ "prompt_source": "sample",
596
+ "prompt_path": "",
597
+ "enable_tag_shuffling": false,
598
+ "tag_delimiter": ",",
599
+ "keep_tags_count": 1,
600
+ "tag_dropout_enable": false,
601
+ "tag_dropout_mode": "FULL",
602
+ "tag_dropout_probability": 0.0,
603
+ "tag_dropout_special_tags_mode": "NONE",
604
+ "tag_dropout_special_tags": "",
605
+ "tag_dropout_special_tags_regex": false,
606
+ "caps_randomize_enable": false,
607
+ "caps_randomize_mode": "capslock, title, first, random",
608
+ "caps_randomize_probability": 0.0,
609
+ "caps_randomize_lowercase": false
610
+ },
611
+ "name": "gayfaceandcock-2048",
612
+ "path": "/workspace/Datasets/r_gayfaceandcock/2048",
613
+ "seed": 634934184,
614
+ "enabled": true,
615
+ "type": "STANDARD",
616
+ "include_subdirectories": false,
617
+ "image_variations": 1,
618
+ "text_variations": 1,
619
+ "balancing": 1.0,
620
+ "balancing_strategy": "REPEATS",
621
+ "loss_weight": 1.0,
622
+ "concept_stats": {}
623
+ },
624
+ {
625
+ "__version": 2,
626
+ "image": {
627
+ "__version": 0,
628
+ "enable_crop_jitter": true,
629
+ "enable_random_flip": false,
630
+ "enable_fixed_flip": false,
631
+ "enable_random_rotate": false,
632
+ "enable_fixed_rotate": false,
633
+ "random_rotate_max_angle": 0.0,
634
+ "enable_random_brightness": false,
635
+ "enable_fixed_brightness": false,
636
+ "random_brightness_max_strength": 0.0,
637
+ "enable_random_contrast": false,
638
+ "enable_fixed_contrast": false,
639
+ "random_contrast_max_strength": 0.0,
640
+ "enable_random_saturation": false,
641
+ "enable_fixed_saturation": false,
642
+ "random_saturation_max_strength": 0.0,
643
+ "enable_random_hue": false,
644
+ "enable_fixed_hue": false,
645
+ "random_hue_max_strength": 0.0,
646
+ "enable_resolution_override": true,
647
+ "resolution_override": "1024",
648
+ "enable_random_circular_mask_shrink": false,
649
+ "enable_random_mask_rotate_crop": false
650
+ },
651
+ "text": {
652
+ "__version": 0,
653
+ "prompt_source": "sample",
654
+ "prompt_path": "",
655
+ "enable_tag_shuffling": false,
656
+ "tag_delimiter": ",",
657
+ "keep_tags_count": 1,
658
+ "tag_dropout_enable": false,
659
+ "tag_dropout_mode": "FULL",
660
+ "tag_dropout_probability": 0.0,
661
+ "tag_dropout_special_tags_mode": "NONE",
662
+ "tag_dropout_special_tags": "",
663
+ "tag_dropout_special_tags_regex": false,
664
+ "caps_randomize_enable": false,
665
+ "caps_randomize_mode": "capslock, title, first, random",
666
+ "caps_randomize_probability": 0.0,
667
+ "caps_randomize_lowercase": false
668
+ },
669
+ "name": "ladybonersgw-1024",
670
+ "path": "/workspace/Datasets/r_ladybonersgw/1408",
671
+ "seed": -39295866,
672
+ "enabled": true,
673
+ "type": "STANDARD",
674
+ "include_subdirectories": false,
675
+ "image_variations": 1,
676
+ "text_variations": 1,
677
+ "balancing": 1.0,
678
+ "balancing_strategy": "REPEATS",
679
+ "loss_weight": 1.0,
680
+ "concept_stats": {}
681
+ },
682
+ {
683
+ "__version": 2,
684
+ "image": {
685
+ "__version": 0,
686
+ "enable_crop_jitter": true,
687
+ "enable_random_flip": false,
688
+ "enable_fixed_flip": false,
689
+ "enable_random_rotate": false,
690
+ "enable_fixed_rotate": false,
691
+ "random_rotate_max_angle": 0.0,
692
+ "enable_random_brightness": false,
693
+ "enable_fixed_brightness": false,
694
+ "random_brightness_max_strength": 0.0,
695
+ "enable_random_contrast": false,
696
+ "enable_fixed_contrast": false,
697
+ "random_contrast_max_strength": 0.0,
698
+ "enable_random_saturation": false,
699
+ "enable_fixed_saturation": false,
700
+ "random_saturation_max_strength": 0.0,
701
+ "enable_random_hue": false,
702
+ "enable_fixed_hue": false,
703
+ "random_hue_max_strength": 0.0,
704
+ "enable_resolution_override": false,
705
+ "resolution_override": "512",
706
+ "enable_random_circular_mask_shrink": false,
707
+ "enable_random_mask_rotate_crop": false
708
+ },
709
+ "text": {
710
+ "__version": 0,
711
+ "prompt_source": "sample",
712
+ "prompt_path": "",
713
+ "enable_tag_shuffling": false,
714
+ "tag_delimiter": ",",
715
+ "keep_tags_count": 1,
716
+ "tag_dropout_enable": false,
717
+ "tag_dropout_mode": "FULL",
718
+ "tag_dropout_probability": 0.0,
719
+ "tag_dropout_special_tags_mode": "NONE",
720
+ "tag_dropout_special_tags": "",
721
+ "tag_dropout_special_tags_regex": false,
722
+ "caps_randomize_enable": false,
723
+ "caps_randomize_mode": "capslock, title, first, random",
724
+ "caps_randomize_probability": 0.0,
725
+ "caps_randomize_lowercase": false
726
+ },
727
+ "name": "ladybonersgw-1536",
728
+ "path": "/workspace/Datasets/r_ladybonersgw/1792",
729
+ "seed": -157702382,
730
+ "enabled": true,
731
+ "type": "STANDARD",
732
+ "include_subdirectories": false,
733
+ "image_variations": 1,
734
+ "text_variations": 1,
735
+ "balancing": 1.0,
736
+ "balancing_strategy": "REPEATS",
737
+ "loss_weight": 1.0,
738
+ "concept_stats": {}
739
+ },
740
+ {
741
+ "__version": 2,
742
+ "image": {
743
+ "__version": 0,
744
+ "enable_crop_jitter": true,
745
+ "enable_random_flip": false,
746
+ "enable_fixed_flip": false,
747
+ "enable_random_rotate": false,
748
+ "enable_fixed_rotate": false,
749
+ "random_rotate_max_angle": 0.0,
750
+ "enable_random_brightness": false,
751
+ "enable_fixed_brightness": false,
752
+ "random_brightness_max_strength": 0.0,
753
+ "enable_random_contrast": false,
754
+ "enable_fixed_contrast": false,
755
+ "random_contrast_max_strength": 0.0,
756
+ "enable_random_saturation": false,
757
+ "enable_fixed_saturation": false,
758
+ "random_saturation_max_strength": 0.0,
759
+ "enable_random_hue": false,
760
+ "enable_fixed_hue": false,
761
+ "random_hue_max_strength": 0.0,
762
+ "enable_resolution_override": false,
763
+ "resolution_override": "512",
764
+ "enable_random_circular_mask_shrink": false,
765
+ "enable_random_mask_rotate_crop": false
766
+ },
767
+ "text": {
768
+ "__version": 0,
769
+ "prompt_source": "sample",
770
+ "prompt_path": "",
771
+ "enable_tag_shuffling": false,
772
+ "tag_delimiter": ",",
773
+ "keep_tags_count": 1,
774
+ "tag_dropout_enable": false,
775
+ "tag_dropout_mode": "FULL",
776
+ "tag_dropout_probability": 0.0,
777
+ "tag_dropout_special_tags_mode": "NONE",
778
+ "tag_dropout_special_tags": "",
779
+ "tag_dropout_special_tags_regex": false,
780
+ "caps_randomize_enable": false,
781
+ "caps_randomize_mode": "capslock, title, first, random",
782
+ "caps_randomize_probability": 0.0,
783
+ "caps_randomize_lowercase": false
784
+ },
785
+ "name": "ladybonersgw-2048",
786
+ "path": "/workspace/Datasets/r_ladybonersgw/2048",
787
+ "seed": -876890768,
788
+ "enabled": true,
789
+ "type": "STANDARD",
790
+ "include_subdirectories": false,
791
+ "image_variations": 1,
792
+ "text_variations": 1,
793
+ "balancing": 1.0,
794
+ "balancing_strategy": "REPEATS",
795
+ "loss_weight": 1.0,
796
+ "concept_stats": {}
797
+ },
798
+ {
799
+ "__version": 2,
800
+ "image": {
801
+ "__version": 0,
802
+ "enable_crop_jitter": true,
803
+ "enable_random_flip": false,
804
+ "enable_fixed_flip": false,
805
+ "enable_random_rotate": false,
806
+ "enable_fixed_rotate": false,
807
+ "random_rotate_max_angle": 0.0,
808
+ "enable_random_brightness": false,
809
+ "enable_fixed_brightness": false,
810
+ "random_brightness_max_strength": 0.0,
811
+ "enable_random_contrast": false,
812
+ "enable_fixed_contrast": false,
813
+ "random_contrast_max_strength": 0.0,
814
+ "enable_random_saturation": false,
815
+ "enable_fixed_saturation": false,
816
+ "random_saturation_max_strength": 0.0,
817
+ "enable_random_hue": false,
818
+ "enable_fixed_hue": false,
819
+ "random_hue_max_strength": 0.0,
820
+ "enable_resolution_override": true,
821
+ "resolution_override": "1024",
822
+ "enable_random_circular_mask_shrink": false,
823
+ "enable_random_mask_rotate_crop": false
824
+ },
825
+ "text": {
826
+ "__version": 0,
827
+ "prompt_source": "sample",
828
+ "prompt_path": "",
829
+ "enable_tag_shuffling": false,
830
+ "tag_delimiter": ",",
831
+ "keep_tags_count": 1,
832
+ "tag_dropout_enable": false,
833
+ "tag_dropout_mode": "FULL",
834
+ "tag_dropout_probability": 0.0,
835
+ "tag_dropout_special_tags_mode": "NONE",
836
+ "tag_dropout_special_tags": "",
837
+ "tag_dropout_special_tags_regex": false,
838
+ "caps_randomize_enable": false,
839
+ "caps_randomize_mode": "capslock, title, first, random",
840
+ "caps_randomize_probability": 0.0,
841
+ "caps_randomize_lowercase": false
842
+ },
843
+ "name": "ratemycock-1024",
844
+ "path": "/workspace/Datasets/r_ratemycock/1408",
845
+ "seed": 159728801,
846
+ "enabled": true,
847
+ "type": "STANDARD",
848
+ "include_subdirectories": false,
849
+ "image_variations": 1,
850
+ "text_variations": 1,
851
+ "balancing": 1.0,
852
+ "balancing_strategy": "REPEATS",
853
+ "loss_weight": 1.0,
854
+ "concept_stats": {
855
+ "file_size": 0,
856
+ "image_count": 0,
857
+ "image_with_mask_count": 0,
858
+ "image_with_caption_count": 0,
859
+ "video_count": 0,
860
+ "video_with_mask_count": 0,
861
+ "video_with_caption_count": 0,
862
+ "mask_count": 0,
863
+ "paired_masks": 0,
864
+ "unpaired_masks": 0,
865
+ "caption_count": 0,
866
+ "subcaption_count": 0,
867
+ "paired_captions": 0,
868
+ "unpaired_captions": 0,
869
+ "processing_time": 0.04560589790344238,
870
+ "directory_count": 1,
871
+ "max_pixels": [
872
+ 0,
873
+ "-",
874
+ "-"
875
+ ],
876
+ "min_pixels": [
877
+ 1000000000,
878
+ "-",
879
+ "-"
880
+ ],
881
+ "avg_pixels": 0,
882
+ "max_length": [
883
+ 0,
884
+ "-"
885
+ ],
886
+ "min_length": [
887
+ 1000000000,
888
+ "-"
889
+ ],
890
+ "avg_length": 0,
891
+ "max_fps": [
892
+ 0,
893
+ "-"
894
+ ],
895
+ "min_fps": [
896
+ 1000000000,
897
+ "-"
898
+ ],
899
+ "avg_fps": 0,
900
+ "max_caption_length": [
901
+ 0,
902
+ "-",
903
+ 0
904
+ ],
905
+ "min_caption_length": [
906
+ 1000000000,
907
+ "-",
908
+ 0
909
+ ],
910
+ "avg_caption_length": [
911
+ 0,
912
+ 0
913
+ ],
914
+ "aspect_buckets": {
915
+ "0.25": 0,
916
+ "0.285714": 0,
917
+ "0.333333": 0,
918
+ "0.4": 0,
919
+ "0.5": 0,
920
+ "0.571429": 0,
921
+ "0.666667": 0,
922
+ "0.8": 0,
923
+ "1.0": 0,
924
+ "1.25": 0,
925
+ "1.5": 0,
926
+ "1.75": 0,
927
+ "2.0": 0,
928
+ "2.5": 0,
929
+ "3.0": 0,
930
+ "3.5": 0,
931
+ "4.0": 0
932
+ },
933
+ "force_cancelled": false
934
+ }
935
+ },
936
+ {
937
+ "__version": 2,
938
+ "image": {
939
+ "__version": 0,
940
+ "enable_crop_jitter": true,
941
+ "enable_random_flip": false,
942
+ "enable_fixed_flip": false,
943
+ "enable_random_rotate": false,
944
+ "enable_fixed_rotate": false,
945
+ "random_rotate_max_angle": 0.0,
946
+ "enable_random_brightness": false,
947
+ "enable_fixed_brightness": false,
948
+ "random_brightness_max_strength": 0.0,
949
+ "enable_random_contrast": false,
950
+ "enable_fixed_contrast": false,
951
+ "random_contrast_max_strength": 0.0,
952
+ "enable_random_saturation": false,
953
+ "enable_fixed_saturation": false,
954
+ "random_saturation_max_strength": 0.0,
955
+ "enable_random_hue": false,
956
+ "enable_fixed_hue": false,
957
+ "random_hue_max_strength": 0.0,
958
+ "enable_resolution_override": false,
959
+ "resolution_override": "512",
960
+ "enable_random_circular_mask_shrink": false,
961
+ "enable_random_mask_rotate_crop": false
962
+ },
963
+ "text": {
964
+ "__version": 0,
965
+ "prompt_source": "sample",
966
+ "prompt_path": "",
967
+ "enable_tag_shuffling": false,
968
+ "tag_delimiter": ",",
969
+ "keep_tags_count": 1,
970
+ "tag_dropout_enable": false,
971
+ "tag_dropout_mode": "FULL",
972
+ "tag_dropout_probability": 0.0,
973
+ "tag_dropout_special_tags_mode": "NONE",
974
+ "tag_dropout_special_tags": "",
975
+ "tag_dropout_special_tags_regex": false,
976
+ "caps_randomize_enable": false,
977
+ "caps_randomize_mode": "capslock, title, first, random",
978
+ "caps_randomize_probability": 0.0,
979
+ "caps_randomize_lowercase": false
980
+ },
981
+ "name": "ratemycock-1536",
982
+ "path": "/workspace/Datasets/r_ratemycock/1792",
983
+ "seed": -249866332,
984
+ "enabled": true,
985
+ "type": "STANDARD",
986
+ "include_subdirectories": false,
987
+ "image_variations": 1,
988
+ "text_variations": 1,
989
+ "balancing": 1.0,
990
+ "balancing_strategy": "REPEATS",
991
+ "loss_weight": 1.0,
992
+ "concept_stats": {}
993
+ },
994
+ {
995
+ "__version": 2,
996
+ "image": {
997
+ "__version": 0,
998
+ "enable_crop_jitter": true,
999
+ "enable_random_flip": false,
1000
+ "enable_fixed_flip": false,
1001
+ "enable_random_rotate": false,
1002
+ "enable_fixed_rotate": false,
1003
+ "random_rotate_max_angle": 0.0,
1004
+ "enable_random_brightness": false,
1005
+ "enable_fixed_brightness": false,
1006
+ "random_brightness_max_strength": 0.0,
1007
+ "enable_random_contrast": false,
1008
+ "enable_fixed_contrast": false,
1009
+ "random_contrast_max_strength": 0.0,
1010
+ "enable_random_saturation": false,
1011
+ "enable_fixed_saturation": false,
1012
+ "random_saturation_max_strength": 0.0,
1013
+ "enable_random_hue": false,
1014
+ "enable_fixed_hue": false,
1015
+ "random_hue_max_strength": 0.0,
1016
+ "enable_resolution_override": false,
1017
+ "resolution_override": "512",
1018
+ "enable_random_circular_mask_shrink": false,
1019
+ "enable_random_mask_rotate_crop": false
1020
+ },
1021
+ "text": {
1022
+ "__version": 0,
1023
+ "prompt_source": "sample",
1024
+ "prompt_path": "",
1025
+ "enable_tag_shuffling": false,
1026
+ "tag_delimiter": ",",
1027
+ "keep_tags_count": 1,
1028
+ "tag_dropout_enable": false,
1029
+ "tag_dropout_mode": "FULL",
1030
+ "tag_dropout_probability": 0.0,
1031
+ "tag_dropout_special_tags_mode": "NONE",
1032
+ "tag_dropout_special_tags": "",
1033
+ "tag_dropout_special_tags_regex": false,
1034
+ "caps_randomize_enable": false,
1035
+ "caps_randomize_mode": "capslock, title, first, random",
1036
+ "caps_randomize_probability": 0.0,
1037
+ "caps_randomize_lowercase": false
1038
+ },
1039
+ "name": "ratemycock-2048",
1040
+ "path": "/workspace/Datasets/r_ratemycock/2048",
1041
+ "seed": -863006310,
1042
+ "enabled": true,
1043
+ "type": "STANDARD",
1044
+ "include_subdirectories": false,
1045
+ "image_variations": 1,
1046
+ "text_variations": 1,
1047
+ "balancing": 1.0,
1048
+ "balancing_strategy": "REPEATS",
1049
+ "loss_weight": 1.0,
1050
+ "concept_stats": {}
1051
+ }
1052
+ ],
1053
+ "aspect_ratio_bucketing": true,
1054
+ "latent_caching": true,
1055
+ "clear_cache_before_training": false,
1056
+ "learning_rate_scheduler": "COSINE",
1057
+ "custom_learning_rate_scheduler": null,
1058
+ "scheduler_params": [],
1059
+ "learning_rate": 0.0002,
1060
+ "learning_rate_warmup_steps": 50.0,
1061
+ "learning_rate_cycles": 1.0,
1062
+ "learning_rate_min_factor": 0.0,
1063
+ "epochs": 20,
1064
+ "batch_size": 1,
1065
+ "gradient_accumulation_steps": 2,
1066
+ "ema": "OFF",
1067
+ "ema_decay": 0.999,
1068
+ "ema_update_step_interval": 5,
1069
+ "dataloader_threads": 1,
1070
+ "train_device": "cuda",
1071
+ "temp_device": "cpu",
1072
+ "train_dtype": "BFLOAT_16",
1073
+ "fallback_train_dtype": "BFLOAT_16",
1074
+ "enable_autocast_cache": true,
1075
+ "only_cache": false,
1076
+ "resolution": "1024,1536",
1077
+ "frames": "25",
1078
+ "mse_strength": 1.0,
1079
+ "mae_strength": 0.0,
1080
+ "log_cosh_strength": 0.0,
1081
+ "huber_strength": 0.0,
1082
+ "huber_delta": 1.0,
1083
+ "vb_loss_strength": 1.0,
1084
+ "loss_weight_fn": "CONSTANT",
1085
+ "loss_weight_strength": 5.0,
1086
+ "dropout_probability": 0.0,
1087
+ "loss_scaler": "NONE",
1088
+ "learning_rate_scaler": "NONE",
1089
+ "clip_grad_norm": 1.0,
1090
+ "offset_noise_weight": 0.0,
1091
+ "generalized_offset_noise": false,
1092
+ "perturbation_noise_weight": 0.03,
1093
+ "rescale_noise_scheduler_to_zero_terminal_snr": false,
1094
+ "force_v_prediction": false,
1095
+ "force_epsilon_prediction": false,
1096
+ "min_noising_strength": 0.0,
1097
+ "max_noising_strength": 1.0,
1098
+ "timestep_distribution": "LOGIT_NORMAL",
1099
+ "noising_weight": 0.0,
1100
+ "noising_bias": 0.0,
1101
+ "timestep_shift": 3.0,
1102
+ "dynamic_timestep_shifting": false,
1103
+ "unet": {
1104
+ "__version": 0,
1105
+ "model_name": "",
1106
+ "include": true,
1107
+ "train": true,
1108
+ "stop_training_after": 0,
1109
+ "stop_training_after_unit": "NEVER",
1110
+ "learning_rate": null,
1111
+ "weight_dtype": "FLOAT_32",
1112
+ "dropout_probability": 0.0,
1113
+ "train_embedding": true,
1114
+ "attention_mask": false,
1115
+ "guidance_scale": 1.0
1116
+ },
1117
+ "prior": {
1118
+ "__version": 0,
1119
+ "model_name": "",
1120
+ "include": true,
1121
+ "train": true,
1122
+ "stop_training_after": 0,
1123
+ "stop_training_after_unit": "NEVER",
1124
+ "learning_rate": null,
1125
+ "weight_dtype": "FLOAT_32",
1126
+ "dropout_probability": 0.0,
1127
+ "train_embedding": true,
1128
+ "attention_mask": false,
1129
+ "guidance_scale": 1.0
1130
+ },
1131
+ "transformer": {
1132
+ "__version": 0,
1133
+ "model_name": "",
1134
+ "include": true,
1135
+ "train": true,
1136
+ "stop_training_after": 0,
1137
+ "stop_training_after_unit": "NEVER",
1138
+ "learning_rate": null,
1139
+ "weight_dtype": "FLOAT_8",
1140
+ "dropout_probability": 0.0,
1141
+ "train_embedding": true,
1142
+ "attention_mask": false,
1143
+ "guidance_scale": 1.0
1144
+ },
1145
+ "quantization": {
1146
+ "__version": 0,
1147
+ "layer_filter": "layers",
1148
+ "layer_filter_preset": "blocks",
1149
+ "layer_filter_regex": false,
1150
+ "svd_dtype": "NONE",
1151
+ "svd_rank": 16,
1152
+ "cache_dir": null
1153
+ },
1154
+ "text_encoder": {
1155
+ "__version": 0,
1156
+ "model_name": "",
1157
+ "include": true,
1158
+ "train": false,
1159
+ "stop_training_after": 30,
1160
+ "stop_training_after_unit": "EPOCH",
1161
+ "learning_rate": null,
1162
+ "weight_dtype": "FLOAT_8",
1163
+ "dropout_probability": 0.0,
1164
+ "train_embedding": true,
1165
+ "attention_mask": false,
1166
+ "guidance_scale": 1.0
1167
+ },
1168
+ "text_encoder_layer_skip": 0,
1169
+ "text_encoder_sequence_length": 512,
1170
+ "text_encoder_2": {
1171
+ "__version": 0,
1172
+ "model_name": "",
1173
+ "include": true,
1174
+ "train": true,
1175
+ "stop_training_after": 30,
1176
+ "stop_training_after_unit": "EPOCH",
1177
+ "learning_rate": null,
1178
+ "weight_dtype": "FLOAT_32",
1179
+ "dropout_probability": 0.0,
1180
+ "train_embedding": true,
1181
+ "attention_mask": false,
1182
+ "guidance_scale": 1.0
1183
+ },
1184
+ "text_encoder_2_layer_skip": 0,
1185
+ "text_encoder_2_sequence_length": 77,
1186
+ "text_encoder_3": {
1187
+ "__version": 0,
1188
+ "model_name": "",
1189
+ "include": true,
1190
+ "train": true,
1191
+ "stop_training_after": 30,
1192
+ "stop_training_after_unit": "EPOCH",
1193
+ "learning_rate": null,
1194
+ "weight_dtype": "FLOAT_32",
1195
+ "dropout_probability": 0.0,
1196
+ "train_embedding": true,
1197
+ "attention_mask": false,
1198
+ "guidance_scale": 1.0
1199
+ },
1200
+ "text_encoder_3_layer_skip": 0,
1201
+ "text_encoder_4": {
1202
+ "__version": 0,
1203
+ "model_name": "",
1204
+ "include": true,
1205
+ "train": true,
1206
+ "stop_training_after": 30,
1207
+ "stop_training_after_unit": "EPOCH",
1208
+ "learning_rate": null,
1209
+ "weight_dtype": "FLOAT_32",
1210
+ "dropout_probability": 0.0,
1211
+ "train_embedding": true,
1212
+ "attention_mask": false,
1213
+ "guidance_scale": 1.0
1214
+ },
1215
+ "text_encoder_4_layer_skip": 0,
1216
+ "vae": {
1217
+ "__version": 0,
1218
+ "model_name": "",
1219
+ "include": true,
1220
+ "train": true,
1221
+ "stop_training_after": null,
1222
+ "stop_training_after_unit": "NEVER",
1223
+ "learning_rate": null,
1224
+ "weight_dtype": "BFLOAT_16",
1225
+ "dropout_probability": 0.0,
1226
+ "train_embedding": true,
1227
+ "attention_mask": false,
1228
+ "guidance_scale": 1.0
1229
+ },
1230
+ "effnet_encoder": {
1231
+ "__version": 0,
1232
+ "model_name": "",
1233
+ "include": true,
1234
+ "train": true,
1235
+ "stop_training_after": null,
1236
+ "stop_training_after_unit": "NEVER",
1237
+ "learning_rate": null,
1238
+ "weight_dtype": "FLOAT_32",
1239
+ "dropout_probability": 0.0,
1240
+ "train_embedding": true,
1241
+ "attention_mask": false,
1242
+ "guidance_scale": 1.0
1243
+ },
1244
+ "decoder": {
1245
+ "__version": 0,
1246
+ "model_name": "",
1247
+ "include": true,
1248
+ "train": true,
1249
+ "stop_training_after": null,
1250
+ "stop_training_after_unit": "NEVER",
1251
+ "learning_rate": null,
1252
+ "weight_dtype": "FLOAT_32",
1253
+ "dropout_probability": 0.0,
1254
+ "train_embedding": true,
1255
+ "attention_mask": false,
1256
+ "guidance_scale": 1.0
1257
+ },
1258
+ "decoder_text_encoder": {
1259
+ "__version": 0,
1260
+ "model_name": "",
1261
+ "include": true,
1262
+ "train": true,
1263
+ "stop_training_after": null,
1264
+ "stop_training_after_unit": "NEVER",
1265
+ "learning_rate": null,
1266
+ "weight_dtype": "FLOAT_32",
1267
+ "dropout_probability": 0.0,
1268
+ "train_embedding": true,
1269
+ "attention_mask": false,
1270
+ "guidance_scale": 1.0
1271
+ },
1272
+ "decoder_vqgan": {
1273
+ "__version": 0,
1274
+ "model_name": "",
1275
+ "include": true,
1276
+ "train": true,
1277
+ "stop_training_after": null,
1278
+ "stop_training_after_unit": "NEVER",
1279
+ "learning_rate": null,
1280
+ "weight_dtype": "FLOAT_32",
1281
+ "dropout_probability": 0.0,
1282
+ "train_embedding": true,
1283
+ "attention_mask": false,
1284
+ "guidance_scale": 1.0
1285
+ },
1286
+ "masked_training": false,
1287
+ "unmasked_probability": 0.1,
1288
+ "unmasked_weight": 0.1,
1289
+ "normalize_masked_area_loss": false,
1290
+ "masked_prior_preservation_weight": 0.0,
1291
+ "custom_conditioning_image": false,
1292
+ "layer_filter": "^(?=.*attention)(?!.*refiner).*,^(?=.*feed_forward)(?!.*refiner).*",
1293
+ "layer_filter_preset": "attn-mlp",
1294
+ "layer_filter_regex": true,
1295
+ "embedding_learning_rate": null,
1296
+ "preserve_embedding_norm": false,
1297
+ "embedding": {
1298
+ "__version": 0,
1299
+ "uuid": "fdeb6425-1327-4993-aa4e-0f3f096c596a",
1300
+ "model_name": "",
1301
+ "placeholder": "<embedding>",
1302
+ "train": true,
1303
+ "stop_training_after": null,
1304
+ "stop_training_after_unit": "NEVER",
1305
+ "token_count": 1,
1306
+ "initial_embedding_text": "*",
1307
+ "is_output_embedding": false
1308
+ },
1309
+ "additional_embeddings": [],
1310
+ "embedding_weight_dtype": "FLOAT_32",
1311
+ "cloud": {
1312
+ "__version": 0,
1313
+ "enabled": false,
1314
+ "type": "RUNPOD",
1315
+ "file_sync": "NATIVE_SCP",
1316
+ "create": true,
1317
+ "name": "OneTrainer",
1318
+ "tensorboard_tunnel": true,
1319
+ "sub_type": "",
1320
+ "gpu_type": "",
1321
+ "volume_size": 100,
1322
+ "min_download": 0,
1323
+ "remote_dir": "/workspace",
1324
+ "huggingface_cache_dir": "/workspace/huggingface_cache",
1325
+ "onetrainer_dir": "/workspace/OneTrainer",
1326
+ "install_cmd": "git clone https://github.com/Nerogar/OneTrainer",
1327
+ "install_onetrainer": true,
1328
+ "update_onetrainer": true,
1329
+ "detach_trainer": false,
1330
+ "run_id": "job1",
1331
+ "download_samples": true,
1332
+ "download_output_model": true,
1333
+ "download_saves": true,
1334
+ "download_backups": false,
1335
+ "download_tensorboard": false,
1336
+ "delete_workspace": false,
1337
+ "on_finish": "NONE",
1338
+ "on_error": "NONE",
1339
+ "on_detached_finish": "NONE",
1340
+ "on_detached_error": "NONE"
1341
+ },
1342
+ "peft_type": "LORA",
1343
+ "lora_model_name": "",
1344
+ "lora_rank": 32,
1345
+ "lora_alpha": 16.0,
1346
+ "lora_decompose": true,
1347
+ "lora_decompose_norm_epsilon": true,
1348
+ "lora_decompose_output_axis": false,
1349
+ "lora_weight_dtype": "BFLOAT_16",
1350
+ "bundle_additional_embeddings": true,
1351
+ "oft_block_size": 32,
1352
+ "oft_coft": false,
1353
+ "coft_eps": 0.0001,
1354
+ "oft_block_share": false,
1355
+ "optimizer": {
1356
+ "__version": 0,
1357
+ "optimizer": "ADAMW_8BIT",
1358
+ "adam_w_mode": false,
1359
+ "alpha": null,
1360
+ "amsgrad": false,
1361
+ "beta1": 0.9,
1362
+ "beta2": 0.999,
1363
+ "beta3": null,
1364
+ "bias_correction": false,
1365
+ "block_wise": true,
1366
+ "capturable": false,
1367
+ "centered": false,
1368
+ "clip_threshold": null,
1369
+ "d0": null,
1370
+ "d_coef": null,
1371
+ "dampening": null,
1372
+ "decay_rate": null,
1373
+ "decouple": false,
1374
+ "differentiable": false,
1375
+ "eps": 1e-08,
1376
+ "eps2": null,
1377
+ "foreach": false,
1378
+ "fsdp_in_use": false,
1379
+ "fused": false,
1380
+ "fused_back_pass": false,
1381
+ "growth_rate": null,
1382
+ "initial_accumulator_value": null,
1383
+ "initial_accumulator": null,
1384
+ "is_paged": false,
1385
+ "log_every": null,
1386
+ "lr_decay": null,
1387
+ "max_unorm": null,
1388
+ "maximize": false,
1389
+ "min_8bit_size": 4096,
1390
+ "quant_block_size": null,
1391
+ "momentum": null,
1392
+ "nesterov": false,
1393
+ "no_prox": false,
1394
+ "optim_bits": 32,
1395
+ "percentile_clipping": 100,
1396
+ "r": null,
1397
+ "relative_step": false,
1398
+ "safeguard_warmup": false,
1399
+ "scale_parameter": false,
1400
+ "stochastic_rounding": true,
1401
+ "use_bias_correction": false,
1402
+ "use_triton": false,
1403
+ "warmup_init": false,
1404
+ "weight_decay": 0.01,
1405
+ "weight_lr_power": null,
1406
+ "decoupled_decay": false,
1407
+ "fixed_decay": false,
1408
+ "rectify": false,
1409
+ "degenerated_to_sgd": false,
1410
+ "k": null,
1411
+ "xi": null,
1412
+ "n_sma_threshold": null,
1413
+ "ams_bound": false,
1414
+ "adanorm": false,
1415
+ "adam_debias": false,
1416
+ "slice_p": null,
1417
+ "cautious": false,
1418
+ "weight_decay_by_lr": true,
1419
+ "prodigy_steps": null,
1420
+ "use_speed": false,
1421
+ "split_groups": true,
1422
+ "split_groups_mean": true,
1423
+ "factored": true,
1424
+ "factored_fp32": true,
1425
+ "use_stableadamw": true,
1426
+ "use_cautious": false,
1427
+ "use_grams": false,
1428
+ "use_adopt": false,
1429
+ "d_limiter": true,
1430
+ "use_schedulefree": true,
1431
+ "use_orthograd": false,
1432
+ "nnmf_factor": false,
1433
+ "orthogonal_gradient": false,
1434
+ "use_atan2": false,
1435
+ "use_AdEMAMix": false,
1436
+ "beta3_ema": null,
1437
+ "alpha_grad": null,
1438
+ "beta1_warmup": null,
1439
+ "min_beta1": null,
1440
+ "Simplified_AdEMAMix": false,
1441
+ "cautious_mask": false,
1442
+ "grams_moment": false,
1443
+ "kourkoutas_beta": false,
1444
+ "k_warmup_steps": null,
1445
+ "schedulefree_c": null,
1446
+ "ns_steps": null,
1447
+ "MuonWithAuxAdam": false,
1448
+ "muon_hidden_layers": null,
1449
+ "muon_adam_regex": false,
1450
+ "muon_adam_lr": null,
1451
+ "muon_te1_adam_lr": null,
1452
+ "muon_te2_adam_lr": null,
1453
+ "muon_adam_config": null,
1454
+ "rms_rescaling": true,
1455
+ "normuon_variant": false,
1456
+ "beta2_normuon": null,
1457
+ "normuon_eps": null,
1458
+ "low_rank_ortho": false,
1459
+ "ortho_rank": null,
1460
+ "accelerated_ns": false,
1461
+ "cautious_wd": false,
1462
+ "approx_mars": false,
1463
+ "kappa_p": null,
1464
+ "auto_kappa_p": false,
1465
+ "compile": false
1466
+ },
1467
+ "optimizer_defaults": {},
1468
+ "sample_definition_file_name": "training_samples/samples.json",
1469
+ "samples": [
1470
+ {
1471
+ "__version": 0,
1472
+ "enabled": true,
1473
+ "prompt": "A high-angle selfie photograph \n captures a light-skinned man in his 30s lying in bed. He is propped up \n against several large, off-white pillows, looking directly at the camera with \n a gentle smile. The man has a bald head with some dark hair on the sides, a \n dark goatee, and a hairy chest and abdomen. His right arm is extended, \n holding the camera, while his left hand is positioned near his groin, \n grasping his semi-erect penis. The setting is intimate and casual, with the \n bed covered in a mix of blue and brown blankets.\\n\\nThe image is a realistic, \n unfiltered photograph with warm, indoor lighting that casts soft shadows \n across his body and the bedding. The dominant color palette is composed of \n neutral tones like cream, tan, and blue, which contrast with his natural skin \n tones and dark body hair. The composition is a medium close-up, focusing on \n his upper body and face, creating a sense of personal connection and \n vulnerability.\\n\\nThe man's physical features are prominent, including thick \n chest hair that tapers down his belly and dense dark hair in his armpit. His \n semi-erect penis is clearly visible, held by his hand. The overall mood of \n the image is relaxed and comfortable, conveying a sense of self-assuredness \n and intimacy. The man's expression is one of contentment, and the informal \n nature of the shot suggests a private moment captured in a personal \n setting.",
1474
+ "negative_prompt": "",
1475
+ "height": 800,
1476
+ "width": 1200,
1477
+ "frames": 1,
1478
+ "length": 10.0,
1479
+ "seed": 1142,
1480
+ "random_seed": false,
1481
+ "diffusion_steps": 20,
1482
+ "cfg_scale": 7.0,
1483
+ "noise_scheduler": "DDIM",
1484
+ "text_encoder_1_layer_skip": 0,
1485
+ "text_encoder_1_sequence_length": null,
1486
+ "text_encoder_2_layer_skip": 0,
1487
+ "text_encoder_2_sequence_length": null,
1488
+ "text_encoder_3_layer_skip": 0,
1489
+ "text_encoder_4_layer_skip": 0,
1490
+ "transformer_attention_mask": false,
1491
+ "force_last_timestep": false,
1492
+ "sample_inpainting": false,
1493
+ "base_image_path": "",
1494
+ "mask_image_path": ""
1495
+ },
1496
+ {
1497
+ "__version": 0,
1498
+ "enabled": true,
1499
+ "prompt": "A full-frontal photograph features \n a man in his 30s with a muscular, hairy physique and a prominent dark \n mustache and goatee. He is seated with his legs spread wide, facing the \n camera directly. His arms are raised, and his hands are tucked behind his \n head, adjusting or holding a camouflage-patterned baseball cap. He has a \n neutral, intense gaze directed toward the viewer. The setting appears to be \n an indoor room, with a grey-patterned curtain visible on the left and a \n framed art piece depicting a deer in a snowy woods on the right wall.\\n\\nThe \n image is a realistic photograph with an amateur, intimate quality. The \n lighting is interior and slightly dim, originating from above and in front of \n the subject, which creates soft shadows under his arms and highlights the \n contours of his torso. The color palette is dominated by natural flesh tones, \n dark body hair, and the neutral grey and brown tones of the background \n elements.\\n\\nThe man is completely nude, displaying full frontal exposure. \n His torso is covered in thick, dark hair that extends from his chest down to \n his stomach and into a dense pubic region. His genitals are centrally framed \n and fully visible; his penis is in a flaccid or semi-erect state, resting \n over his scrotum. His legs are spread wide, showing hairy thighs and more of \n the groin area. The low-angle composition emphasizes his body and genitals, \n conveying a sense of exhibitionism and confidence. The combination of his \n pose, facial expression, and the rustic-themed decor in the background \n suggests a deliberate presentation of raw, masculine energy.",
1500
+ "negative_prompt": "",
1501
+ "height": 1536,
1502
+ "width": 1536,
1503
+ "frames": 1,
1504
+ "length": 10.0,
1505
+ "seed": 2242,
1506
+ "random_seed": false,
1507
+ "diffusion_steps": 20,
1508
+ "cfg_scale": 7.0,
1509
+ "noise_scheduler": "DDIM",
1510
+ "text_encoder_1_layer_skip": 0,
1511
+ "text_encoder_1_sequence_length": null,
1512
+ "text_encoder_2_layer_skip": 0,
1513
+ "text_encoder_2_sequence_length": null,
1514
+ "text_encoder_3_layer_skip": 0,
1515
+ "text_encoder_4_layer_skip": 0,
1516
+ "transformer_attention_mask": false,
1517
+ "force_last_timestep": false,
1518
+ "sample_inpainting": false,
1519
+ "base_image_path": "",
1520
+ "mask_image_path": ""
1521
+ },
1522
+ {
1523
+ "__version": 0,
1524
+ "enabled": true,
1525
+ "prompt": "This realistic photograph features \n a fit, light-skinned adult man in his 30s with short, sandy-blond hair and a \n well-defined, athletic physique. He is standing in a bathroom, taking a \n mirror selfie. He holds a white smartphone in his right hand while using his \n left hand to pull down the waistband of his blue Jockey boxer briefs, \n exposing his lower torso and groin area. The background shows a portion of a \n bedroom with a neatly made bed and a gray textured shower curtain to his \n right.\\n\\nThe image is a medium-framed shot captured with clear, even indoor \n lighting that emphasizes the contours of the man's chest and abdominal \n muscles. The color palette is composed of neutral tones, including the beige \n bathroom walls, the off-white bedroom beyond the doorway, and the man's fair \n skin. The overall mood is casual and intimate, typical of a personal \n self-portrait.\\n\\nIn a state of partial nudity, the man has pulled his \n underwear down to reveal his pubic region. He has a patch of reddish-brown \n pubic hair. His flaccid penis is fully visible, resting over the lowered \n waistband of his underwear. The exposure is deliberate and presented in a \n straightforward manner.\\n\\nThe man maintains a neutral, somewhat focused \n expression as he looks toward the mirror to frame his shot. The setting is a \n standard domestic environment, and the inclusion of the bedroom in the \n background enhances the private, personal nature of the photograph. The image \n captures a moment of physical self-documentation in a private space.",
1526
+ "negative_prompt": "",
1527
+ "height": 2400,
1528
+ "width": 1600,
1529
+ "frames": 1,
1530
+ "length": 10.0,
1531
+ "seed": 3342,
1532
+ "random_seed": false,
1533
+ "diffusion_steps": 20,
1534
+ "cfg_scale": 7.0,
1535
+ "noise_scheduler": "DDIM",
1536
+ "text_encoder_1_layer_skip": 0,
1537
+ "text_encoder_1_sequence_length": null,
1538
+ "text_encoder_2_layer_skip": 0,
1539
+ "text_encoder_2_sequence_length": null,
1540
+ "text_encoder_3_layer_skip": 0,
1541
+ "text_encoder_4_layer_skip": 0,
1542
+ "transformer_attention_mask": false,
1543
+ "force_last_timestep": false,
1544
+ "sample_inpainting": false,
1545
+ "base_image_path": "",
1546
+ "mask_image_path": ""
1547
+ }
1548
+ ],
1549
+ "sample_after": 200,
1550
+ "sample_after_unit": "STEP",
1551
+ "sample_skip_first": 0,
1552
+ "sample_image_format": "JPG",
1553
+ "sample_video_format": "MP4",
1554
+ "sample_audio_format": "MP3",
1555
+ "samples_to_tensorboard": true,
1556
+ "non_ema_sampling": true,
1557
+ "backup_after": 30,
1558
+ "backup_after_unit": "MINUTE",
1559
+ "rolling_backup": false,
1560
+ "rolling_backup_count": 3,
1561
+ "backup_before_save": true,
1562
+ "save_every": 200,
1563
+ "save_every_unit": "STEP",
1564
+ "save_skip_first": 0,
1565
+ "save_filename_prefix": "reddit-2-zib"
1566
+ }