misc-models / README.md
kohya-ss's picture
Update README.md
01c8fd3 verified
|
raw
history blame
2.39 kB
## flux-hasui-lora-d4-sigmoid-raw-gs1.0.safetensors
Experimental LoRA for FLUX.1 dev.
Trained with `sd-scripts` (Aug. 11) `sd3` branch. __NOTE:__ This settings requires > 26GB VRAM. Please add `--fp8_base` to enable fp8 training to reduce VRAM usage.
```
accelerate launch --mixed_precision bf16 --num_cpu_threads_per_process 1 flux_train_network.py --pretrained_model_name_or_path flux1/flux1-dev.sft --clip_l sd3/clip_l.safetensors --t5xxl sd3/t5xxl_fp16.safetensors --ae flux1/ae_dev.sft --cache_latents_to_disk --save_model_as safetensors --sdpa --persistent_data_loader_workers --max_data_loader_n_workers 2 --seed 42 --gradient_checkpointing --mixed_precision bf16 --save_precision bf16 --network_module networks.lora_flux --network_dim 4 --optimizer_type adamw8bit --learning_rate 1e-3 --network_train_unet_only --cache_text_encoder_outputs --cache_text_encoder_outputs_to_disk --highvram --max_train_epochs 4 --save_every_n_epochs 1 --dataset_config hasui_1024_bs1.toml --output_dir flux/lora --output_name lora-name --timestep_sampling sigmoid --model_prediction_type raw --guidance_scale 1.0
```
.toml is below.
```.toml
[general]
flip_aug = true
color_aug = false
[[datasets]]
enable_bucket = true
resolution = [1024,1024]
bucket_reso_steps = 64
max_bucket_reso = 2048
min_bucket_reso = 128
bucket_no_upscale = false
batch_size = 1
random_crop = false
shuffle_caption = false
[[datasets.subsets]]
image_dir = "path/to/train/images"
num_repeats = 1
caption_extension = ".txt"
```
## sdxl-negprompt8-v1m.safetensors
Negative embeddings for sdxl. Num vectors per token = 8
## stable-cascade-c-lora-hasui-v02.safetensors
Sample of LoRA for Stable Cascade Stage C.
Feb 22, 2024 Update: Fixed a bug that LoRA is not applied to some modules (to_q/k/v and to_out) in Attention.
__This is an experimental model, so the format of the weights may change in the future.__
- a painting of an anthropomorphic penguin sitting in a cafe reading a book and having a coffee --w 1024 --h 1024 --d 1
![sample1](penguin.png)
- a painting of japanese shrine in winter with snowfall --w 832 --h 1152 --d 1234
![sample2](shrine.png)
This model is trained with 169 images with captions. U-Net only, dim=4, conv_dim=4, alpha=1, lr=1e-3, 4 epochs, mixed precision bf16, 8bit AdamW, batch size 8, resolution 1024x1024 with aspect ratio bucketing. VRAM usage is approximately 22 GB.