AdwolfCzar commited on
Commit
d920a8d
·
verified ·
1 Parent(s): 8b9f815

Upload trainero_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. trainero_config.json +47 -0
trainero_config.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "flux-klein",
3
+ "mode": "style-rush",
4
+ "trigger": "anime loven",
5
+ "schedule": {
6
+ "num_repeats": 1,
7
+ "epochs": 5,
8
+ "save_every_n_epochs": 1
9
+ },
10
+ "style_rush": {
11
+ "pairs": 50,
12
+ "refused": 6,
13
+ "failed": 0,
14
+ "cost": 0.417955
15
+ },
16
+ "config": {
17
+ "dit": "/workspace/models/flux-klein/flux-2-klein-base-9b.safetensors",
18
+ "vae": "/workspace/models/flux-klein/ae.safetensors",
19
+ "text_encoder": "/workspace/models/flux-klein/text_encoder/model-00001-of-00004.safetensors",
20
+ "model_version": "klein-base-9b",
21
+ "network_dim": 128,
22
+ "network_alpha": 128,
23
+ "learning_rate": 0.0001,
24
+ "optimizer_type": "adamw8bit",
25
+ "lr_scheduler": "cosine",
26
+ "timestep_sampling": "flux2_shift",
27
+ "weighting_scheme": "none",
28
+ "fp8_base": true,
29
+ "fp8_scaled": true,
30
+ "network_module": "networks.lora_flux_2",
31
+ "dataset_config": "/workspace/trainero_projects/lovstyleanimex/dataset.toml",
32
+ "mixed_precision": "bf16",
33
+ "gradient_checkpointing": true,
34
+ "max_train_epochs": 5,
35
+ "save_every_n_epochs": 1,
36
+ "seed": 42,
37
+ "output_dir": "/workspace/trainero_projects/lovstyleanimex/output",
38
+ "output_name": "lovstyleanimex",
39
+ "max_data_loader_n_workers": 2,
40
+ "persistent_data_loader_workers": true,
41
+ "sample_prompts": "/workspace/trainero_projects/lovstyleanimex/sample_prompts.txt",
42
+ "sample_every_n_epochs": 1,
43
+ "sample_at_first": true,
44
+ "sdpa": true,
45
+ "lr_warmup_steps": 77
46
+ }
47
+ }