CoolFace
Modelpublic

Aryan-21/base-sd2-id-89

sourceHugging Faceotherupdated 2y agoView on Hugging Face
0likes18downloads
train.sh34 linesDownload Raw Back to root
1accelerate 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 "/home/genai/Aryan/facecode/Generative/fluxgym/models/unet/flux1-dev.sft" \6  --clip_l "/home/genai/Aryan/facecode/Generative/fluxgym/models/clip/clip_l.safetensors" \7  --t5xxl "/home/genai/Aryan/facecode/Generative/fluxgym/models/clip/t5xxl_fp16.safetensors" \8  --ae "/home/genai/Aryan/facecode/Generative/fluxgym/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 \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 "/home/genai/Aryan/facecode/Generative/fluxgym/outputs/base-sd2-id-89/dataset.toml" \28  --output_dir "/home/genai/Aryan/facecode/Generative/fluxgym/outputs/base-sd2-id-89" \29  --output_name base-sd2-id-89 \30  --timestep_sampling shift \31  --discrete_flow_shift 3.1582 \32  --model_prediction_type raw \33  --guidance_scale 1 \34  --loss_type l2 \