Upload 7 files
Browse files- dataset.toml +14 -0
- juice-orb-000004.safetensors +3 -0
- juice-orb-000008.safetensors +3 -0
- juice-orb-000012.safetensors +3 -0
- juice-orb-000016.safetensors +3 -0
- juice-orb.safetensors +3 -0
- train.bat +39 -0
dataset.toml
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[general]
|
2 |
+
shuffle_caption = false
|
3 |
+
caption_extension = '.txt'
|
4 |
+
keep_tokens = 1
|
5 |
+
|
6 |
+
[[datasets]]
|
7 |
+
resolution = 512
|
8 |
+
batch_size = 1
|
9 |
+
keep_tokens = 1
|
10 |
+
|
11 |
+
[[datasets.subsets]]
|
12 |
+
image_dir = 'D:\pinokio\api\fluxgym.git\datasets\juice-orb'
|
13 |
+
class_tokens = '0rb'
|
14 |
+
num_repeats = 10
|
juice-orb-000004.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3867f2f2aeabd44370ac98be8b9d673183586fad3bee3cca0b8b164dfabebb02
|
3 |
+
size 36346588
|
juice-orb-000008.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:57d920adc3b47a1b74ab9cecdc5b050b3cf0e5cb11befbbe7a4f7ff5222da0bb
|
3 |
+
size 36346588
|
juice-orb-000012.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0f7b9daeb33ddeea9223f0576881a7e6a88c76848acdcaaa83a47c63ffdecf44
|
3 |
+
size 36346588
|
juice-orb-000016.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9d696748fa375f16af8051cdb36cbb09faddc61e19922fed5434fb0fccf0311c
|
3 |
+
size 36346588
|
juice-orb.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3c09a1b31689afc7fc153c8081ab5ae4c373332f9db548b36f78d4cce9efb28f
|
3 |
+
size 36346588
|
train.bat
ADDED
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
accelerate launch ^
|
2 |
+
--mixed_precision bf16 ^
|
3 |
+
--num_cpu_threads_per_process 1 ^
|
4 |
+
sd-scripts/flux_train_network.py ^
|
5 |
+
--pretrained_model_name_or_path "D:\pinokio\api\fluxgym.git\models\unet\flux1-dev.sft" ^
|
6 |
+
--clip_l "D:\pinokio\api\fluxgym.git\models\clip\clip_l.safetensors" ^
|
7 |
+
--t5xxl "D:\pinokio\api\fluxgym.git\models\clip\t5xxl_fp16.safetensors" ^
|
8 |
+
--ae "D:\pinokio\api\fluxgym.git\models\vae\ae.sft" ^
|
9 |
+
--cache_latents_to_disk ^
|
10 |
+
--save_model_as safetensors ^
|
11 |
+
--sdpa --persistent_data_loader_workers ^
|
12 |
+
--max_data_loader_n_workers 2 ^
|
13 |
+
--seed 42 ^
|
14 |
+
--gradient_checkpointing ^
|
15 |
+
--mixed_precision bf16 ^
|
16 |
+
--save_precision bf16 ^
|
17 |
+
--network_module networks.lora_flux ^
|
18 |
+
--network_dim 8 ^
|
19 |
+
--optimizer_type adafactor ^
|
20 |
+
--optimizer_args "relative_step=False" "scale_parameter=False" "warmup_init=False" ^
|
21 |
+
--split_mode ^
|
22 |
+
--network_args "train_blocks=single" ^
|
23 |
+
--lr_scheduler constant_with_warmup ^
|
24 |
+
--max_grad_norm 0.0 ^--sample_prompts="D:\pinokio\api\fluxgym.git\outputs\juice-orb\sample_prompts.txt" --sample_every_n_steps="150" ^
|
25 |
+
--learning_rate 6e-4 ^
|
26 |
+
--cache_text_encoder_outputs ^
|
27 |
+
--cache_text_encoder_outputs_to_disk ^
|
28 |
+
--fp8_base ^
|
29 |
+
--highvram ^
|
30 |
+
--max_train_epochs 20 ^
|
31 |
+
--save_every_n_epochs 4 ^
|
32 |
+
--dataset_config "D:\pinokio\api\fluxgym.git\outputs\juice-orb\dataset.toml" ^
|
33 |
+
--output_dir "D:\pinokio\api\fluxgym.git\outputs\juice-orb" ^
|
34 |
+
--output_name juice-orb ^
|
35 |
+
--timestep_sampling shift ^
|
36 |
+
--discrete_flow_shift 3.1582 ^
|
37 |
+
--model_prediction_type raw ^
|
38 |
+
--guidance_scale 1 ^
|
39 |
+
--loss_type l2 ^
|