KamCastle commited on
Commit
9f9fcb2
1 Parent(s): ba24915

Upload 2 files

Browse files
Files changed (2) hide show
  1. brkcat.safetensors +3 -0
  2. masscompute.json +378 -0
brkcat.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca7b2c78d296e5f8188d1504fc0c99a076f39da35a62051031afe9c8b7b76f02
3
+ size 6938084280
masscompute.json ADDED
@@ -0,0 +1,378 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version": 3,
3
+ "training_method": "FINE_TUNE",
4
+ "model_type": "STABLE_DIFFUSION_XL_10_BASE",
5
+ "debug_mode": false,
6
+ "debug_dir": "debug",
7
+ "workspace_dir": "/home/Ubuntu/Desktop/onetrainer_workspace",
8
+ "cache_dir": "/home/Ubuntu/Desktop/onetrainer_workspace/cache",
9
+ "tensorboard": false,
10
+ "tensorboard_expose": false,
11
+ "continue_last_backup": false,
12
+ "include_train_config": "NONE",
13
+ "base_model_name": "/home/Ubuntu/Desktop/1T/juggernautXL_juggernautX.safetensors",
14
+ "weight_dtype": "BFLOAT_16",
15
+ "output_dtype": "BFLOAT_16",
16
+ "output_model_format": "SAFETENSORS",
17
+ "output_model_destination": "/home/Ubuntu/Desktop/onetrainer_workspace/brkcat.safetensors",
18
+ "gradient_checkpointing": true,
19
+ "force_circular_padding": false,
20
+ "concept_file_name": "training_concepts/concepts.json",
21
+ "concepts": [
22
+ {
23
+ "__version": 1,
24
+ "image": {
25
+ "__version": 0,
26
+ "enable_crop_jitter": false,
27
+ "enable_random_flip": false,
28
+ "enable_fixed_flip": false,
29
+ "enable_random_rotate": false,
30
+ "enable_fixed_rotate": false,
31
+ "random_rotate_max_angle": 0.0,
32
+ "enable_random_brightness": false,
33
+ "enable_fixed_brightness": false,
34
+ "random_brightness_max_strength": 0.0,
35
+ "enable_random_contrast": false,
36
+ "enable_fixed_contrast": false,
37
+ "random_contrast_max_strength": 0.0,
38
+ "enable_random_saturation": false,
39
+ "enable_fixed_saturation": false,
40
+ "random_saturation_max_strength": 0.0,
41
+ "enable_random_hue": false,
42
+ "enable_fixed_hue": false,
43
+ "random_hue_max_strength": 0.0,
44
+ "enable_resolution_override": false,
45
+ "resolution_override": "512"
46
+ },
47
+ "text": {
48
+ "__version": 0,
49
+ "prompt_source": "concept",
50
+ "prompt_path": "/home/Ubuntu/Desktop/1T/brk cat.txt",
51
+ "enable_tag_shuffling": false,
52
+ "tag_delimiter": ",",
53
+ "keep_tags_count": 1
54
+ },
55
+ "name": "training",
56
+ "path": "/home/Ubuntu/Desktop/1T/brk1024",
57
+ "seed": 866513600,
58
+ "enabled": true,
59
+ "include_subdirectories": false,
60
+ "image_variations": 1,
61
+ "text_variations": 1,
62
+ "balancing": 1.0,
63
+ "balancing_strategy": "REPEATS",
64
+ "loss_weight": 1.0
65
+ },
66
+ {
67
+ "__version": 1,
68
+ "image": {
69
+ "__version": 0,
70
+ "enable_crop_jitter": false,
71
+ "enable_random_flip": false,
72
+ "enable_fixed_flip": false,
73
+ "enable_random_rotate": false,
74
+ "enable_fixed_rotate": false,
75
+ "random_rotate_max_angle": 0.0,
76
+ "enable_random_brightness": false,
77
+ "enable_fixed_brightness": false,
78
+ "random_brightness_max_strength": 0.0,
79
+ "enable_random_contrast": false,
80
+ "enable_fixed_contrast": false,
81
+ "random_contrast_max_strength": 0.0,
82
+ "enable_random_saturation": false,
83
+ "enable_fixed_saturation": false,
84
+ "random_saturation_max_strength": 0.0,
85
+ "enable_random_hue": false,
86
+ "enable_fixed_hue": false,
87
+ "random_hue_max_strength": 0.0,
88
+ "enable_resolution_override": false,
89
+ "resolution_override": "512"
90
+ },
91
+ "text": {
92
+ "__version": 0,
93
+ "prompt_source": "concept",
94
+ "prompt_path": "/home/Ubuntu/Desktop/1T/cat.txt",
95
+ "enable_tag_shuffling": false,
96
+ "tag_delimiter": ",",
97
+ "keep_tags_count": 1
98
+ },
99
+ "name": "reg",
100
+ "path": "/home/Ubuntu/Desktop/1T/cat-unsplash1024x1024",
101
+ "seed": 920357731,
102
+ "enabled": true,
103
+ "include_subdirectories": false,
104
+ "image_variations": 1,
105
+ "text_variations": 1,
106
+ "balancing": 0.02,
107
+ "balancing_strategy": "REPEATS",
108
+ "loss_weight": 1.0
109
+ }
110
+ ],
111
+ "circular_mask_generation": false,
112
+ "random_rotate_and_crop": false,
113
+ "aspect_ratio_bucketing": false,
114
+ "latent_caching": true,
115
+ "clear_cache_before_training": false,
116
+ "learning_rate_scheduler": "CONSTANT",
117
+ "learning_rate": 1e-05,
118
+ "learning_rate_warmup_steps": 200,
119
+ "learning_rate_cycles": 1,
120
+ "epochs": 200,
121
+ "batch_size": 1,
122
+ "gradient_accumulation_steps": 1,
123
+ "ema": "OFF",
124
+ "ema_decay": 0.999,
125
+ "ema_update_step_interval": 1,
126
+ "dataloader_threads": 2,
127
+ "train_device": "cuda",
128
+ "temp_device": "cpu",
129
+ "train_dtype": "BFLOAT_16",
130
+ "fallback_train_dtype": "FLOAT_32",
131
+ "enable_autocast_cache": true,
132
+ "only_cache": false,
133
+ "resolution": "1024",
134
+ "attention_mechanism": "DEFAULT",
135
+ "align_prop": false,
136
+ "align_prop_probability": 0.1,
137
+ "align_prop_loss": "AESTHETIC",
138
+ "align_prop_weight": 0.01,
139
+ "align_prop_steps": 20,
140
+ "align_prop_truncate_steps": 0.5,
141
+ "align_prop_cfg_scale": 7.0,
142
+ "mse_strength": 1.0,
143
+ "mae_strength": 0.0,
144
+ "vb_loss_strength": 1.0,
145
+ "loss_weight_fn": "CONSTANT",
146
+ "loss_weight_strength": 5.0,
147
+ "dropout_probability": 0.0,
148
+ "loss_scaler": "NONE",
149
+ "learning_rate_scaler": "NONE",
150
+ "offset_noise_weight": 0.0,
151
+ "perturbation_noise_weight": 0.0,
152
+ "rescale_noise_scheduler_to_zero_terminal_snr": false,
153
+ "force_v_prediction": false,
154
+ "force_epsilon_prediction": false,
155
+ "min_noising_strength": 0.0,
156
+ "max_noising_strength": 1.0,
157
+ "noising_weight": 0.0,
158
+ "noising_bias": 0.5,
159
+ "unet": {
160
+ "__version": 0,
161
+ "model_name": "",
162
+ "train": true,
163
+ "stop_training_after": 10000,
164
+ "stop_training_after_unit": "EPOCH",
165
+ "learning_rate": 1e-05,
166
+ "weight_dtype": "BFLOAT_16"
167
+ },
168
+ "prior": {
169
+ "__version": 0,
170
+ "model_name": "",
171
+ "train": true,
172
+ "stop_training_after": 10000,
173
+ "stop_training_after_unit": "EPOCH",
174
+ "learning_rate": null,
175
+ "weight_dtype": "NONE"
176
+ },
177
+ "text_encoder": {
178
+ "__version": 0,
179
+ "model_name": "",
180
+ "train": true,
181
+ "stop_training_after": 250,
182
+ "stop_training_after_unit": "EPOCH",
183
+ "learning_rate": 3e-06,
184
+ "weight_dtype": "BFLOAT_16"
185
+ },
186
+ "text_encoder_layer_skip": 0,
187
+ "text_encoder_2": {
188
+ "__version": 0,
189
+ "model_name": "",
190
+ "train": false,
191
+ "stop_training_after": 30,
192
+ "stop_training_after_unit": "EPOCH",
193
+ "learning_rate": null,
194
+ "weight_dtype": "BFLOAT_16"
195
+ },
196
+ "text_encoder_2_layer_skip": 0,
197
+ "vae": {
198
+ "__version": 0,
199
+ "model_name": "",
200
+ "train": true,
201
+ "stop_training_after": null,
202
+ "stop_training_after_unit": "NEVER",
203
+ "learning_rate": null,
204
+ "weight_dtype": "FLOAT_32"
205
+ },
206
+ "effnet_encoder": {
207
+ "__version": 0,
208
+ "model_name": "",
209
+ "train": true,
210
+ "stop_training_after": null,
211
+ "stop_training_after_unit": "NEVER",
212
+ "learning_rate": null,
213
+ "weight_dtype": "NONE"
214
+ },
215
+ "decoder": {
216
+ "__version": 0,
217
+ "model_name": "",
218
+ "train": true,
219
+ "stop_training_after": null,
220
+ "stop_training_after_unit": "NEVER",
221
+ "learning_rate": null,
222
+ "weight_dtype": "NONE"
223
+ },
224
+ "decoder_text_encoder": {
225
+ "__version": 0,
226
+ "model_name": "",
227
+ "train": true,
228
+ "stop_training_after": null,
229
+ "stop_training_after_unit": "NEVER",
230
+ "learning_rate": null,
231
+ "weight_dtype": "NONE"
232
+ },
233
+ "decoder_vqgan": {
234
+ "__version": 0,
235
+ "model_name": "",
236
+ "train": true,
237
+ "stop_training_after": null,
238
+ "stop_training_after_unit": "NEVER",
239
+ "learning_rate": null,
240
+ "weight_dtype": "NONE"
241
+ },
242
+ "masked_training": false,
243
+ "unmasked_probability": 0.1,
244
+ "unmasked_weight": 0.1,
245
+ "normalize_masked_area_loss": false,
246
+ "embedding_learning_rate": null,
247
+ "preserve_embedding_norm": false,
248
+ "embedding": {
249
+ "__version": 0,
250
+ "uuid": "be454e9c-068a-4d34-b0e4-10cf54461eee",
251
+ "model_name": "",
252
+ "placeholder": "<embedding>",
253
+ "train": true,
254
+ "stop_training_after": null,
255
+ "stop_training_after_unit": "NEVER",
256
+ "token_count": 1,
257
+ "initial_embedding_text": "*"
258
+ },
259
+ "additional_embeddings": [],
260
+ "embedding_weight_dtype": "FLOAT_32",
261
+ "lora_model_name": "",
262
+ "lora_rank": 16,
263
+ "lora_alpha": 1.0,
264
+ "lora_weight_dtype": "FLOAT_32",
265
+ "optimizer": {
266
+ "__version": 0,
267
+ "optimizer": "ADAFACTOR",
268
+ "adam_w_mode": false,
269
+ "alpha": null,
270
+ "amsgrad": false,
271
+ "beta1": null,
272
+ "beta2": null,
273
+ "beta3": null,
274
+ "bias_correction": false,
275
+ "block_wise": false,
276
+ "capturable": false,
277
+ "centered": false,
278
+ "clip_threshold": 1.0,
279
+ "d0": null,
280
+ "d_coef": null,
281
+ "dampening": null,
282
+ "decay_rate": -0.8,
283
+ "decouple": false,
284
+ "differentiable": false,
285
+ "eps": 1e-30,
286
+ "eps2": 0.001,
287
+ "foreach": false,
288
+ "fsdp_in_use": false,
289
+ "fused": false,
290
+ "fused_back_pass": false,
291
+ "growth_rate": null,
292
+ "initial_accumulator_value": null,
293
+ "is_paged": false,
294
+ "log_every": null,
295
+ "lr_decay": null,
296
+ "max_unorm": null,
297
+ "maximize": false,
298
+ "min_8bit_size": null,
299
+ "momentum": null,
300
+ "nesterov": false,
301
+ "no_prox": false,
302
+ "optim_bits": null,
303
+ "percentile_clipping": null,
304
+ "relative_step": false,
305
+ "safeguard_warmup": false,
306
+ "scale_parameter": false,
307
+ "stochastic_rounding": false,
308
+ "use_bias_correction": false,
309
+ "use_triton": false,
310
+ "warmup_init": false,
311
+ "weight_decay": 0.01
312
+ },
313
+ "optimizer_defaults": {
314
+ "ADAFACTOR": {
315
+ "__version": 0,
316
+ "optimizer": "ADAFACTOR",
317
+ "adam_w_mode": false,
318
+ "alpha": null,
319
+ "amsgrad": false,
320
+ "beta1": null,
321
+ "beta2": null,
322
+ "beta3": null,
323
+ "bias_correction": false,
324
+ "block_wise": false,
325
+ "capturable": false,
326
+ "centered": false,
327
+ "clip_threshold": 1.0,
328
+ "d0": null,
329
+ "d_coef": null,
330
+ "dampening": null,
331
+ "decay_rate": -0.8,
332
+ "decouple": false,
333
+ "differentiable": false,
334
+ "eps": 1e-30,
335
+ "eps2": 0.001,
336
+ "foreach": false,
337
+ "fsdp_in_use": false,
338
+ "fused": false,
339
+ "fused_back_pass": false,
340
+ "growth_rate": null,
341
+ "initial_accumulator_value": null,
342
+ "is_paged": false,
343
+ "log_every": null,
344
+ "lr_decay": null,
345
+ "max_unorm": null,
346
+ "maximize": false,
347
+ "min_8bit_size": null,
348
+ "momentum": null,
349
+ "nesterov": false,
350
+ "no_prox": false,
351
+ "optim_bits": null,
352
+ "percentile_clipping": null,
353
+ "relative_step": false,
354
+ "safeguard_warmup": false,
355
+ "scale_parameter": false,
356
+ "stochastic_rounding": false,
357
+ "use_bias_correction": false,
358
+ "use_triton": false,
359
+ "warmup_init": false,
360
+ "weight_decay": 0.01
361
+ }
362
+ },
363
+ "sample_definition_file_name": "training_samples/samples.json",
364
+ "samples": [],
365
+ "sample_after": 10,
366
+ "sample_after_unit": "MINUTE",
367
+ "sample_image_format": "JPG",
368
+ "samples_to_tensorboard": false,
369
+ "non_ema_sampling": false,
370
+ "backup_after": 30,
371
+ "backup_after_unit": "NEVER",
372
+ "rolling_backup": false,
373
+ "rolling_backup_count": 3,
374
+ "backup_before_save": false,
375
+ "save_after": 50,
376
+ "save_after_unit": "EPOCH",
377
+ "save_filename_prefix": "brkcat"
378
+ }