VHKE commited on
Commit
ae4727c
·
verified ·
1 Parent(s): 588c682

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. dataset.toml +14 -0
  2. sample_prompts.txt +1 -0
  3. train.bat +34 -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 = 'C:\pinokio\api\fluxgym.git\datasets\open-slippers'
13
+ class_tokens = 'open slippers'
14
+ num_repeats = 10
sample_prompts.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ open slippers
train.bat ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 "C:\pinokio\api\fluxgym.git\models\unet\flux1-dev.sft" ^
6
+ --clip_l "C:\pinokio\api\fluxgym.git\models\clip\clip_l.safetensors" ^
7
+ --t5xxl "C:\pinokio\api\fluxgym.git\models\clip\t5xxl_fp16.safetensors" ^
8
+ --ae "C:\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 4 ^
19
+ --optimizer_type adamw8bit ^--sample_prompts="C:\pinokio\api\fluxgym.git\outputs\open-slippers\sample_prompts.txt" --sample_every_n_steps="500" ^
20
+ --learning_rate 8e-4 ^
21
+ --cache_text_encoder_outputs ^
22
+ --cache_text_encoder_outputs_to_disk ^
23
+ --fp8_base ^
24
+ --highvram ^
25
+ --max_train_epochs 16 ^
26
+ --save_every_n_epochs 4 ^
27
+ --dataset_config "C:\pinokio\api\fluxgym.git\outputs\open-slippers\dataset.toml" ^
28
+ --output_dir "C:\pinokio\api\fluxgym.git\outputs\open-slippers" ^
29
+ --output_name open-slippers ^
30
+ --timestep_sampling shift ^
31
+ --discrete_flow_shift 3.1582 ^
32
+ --model_prediction_type raw ^
33
+ --guidance_scale 1 ^
34
+ --loss_type l2 ^