Text-to-Image
Diffusers
TensorBoard
diffusers-training
lora
flux2-klein
flux2-klein-diffusers
template:sd-lora
Instructions to use Retoe/shatfine3 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Retoe/shatfine3 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("fill-in-base-model", dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("Retoe/shatfine3") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
End of training
Browse files- README.md +9 -0
- checkpoint-1000/optimizer.bin +3 -0
- checkpoint-1000/pytorch_lora_weights.safetensors +3 -0
- checkpoint-1000/random_states_0.pkl +3 -0
- checkpoint-1000/scheduler.bin +3 -0
- checkpoint-1250/optimizer.bin +3 -0
- checkpoint-1250/pytorch_lora_weights.safetensors +3 -0
- checkpoint-1250/random_states_0.pkl +3 -0
- checkpoint-1250/scheduler.bin +3 -0
- checkpoint-250/optimizer.bin +3 -0
- checkpoint-250/pytorch_lora_weights.safetensors +3 -0
- checkpoint-250/random_states_0.pkl +3 -0
- checkpoint-250/scheduler.bin +3 -0
- checkpoint-500/optimizer.bin +3 -0
- checkpoint-500/pytorch_lora_weights.safetensors +3 -0
- checkpoint-500/random_states_0.pkl +3 -0
- checkpoint-500/scheduler.bin +3 -0
- checkpoint-750/optimizer.bin +3 -0
- checkpoint-750/pytorch_lora_weights.safetensors +3 -0
- checkpoint-750/random_states_0.pkl +3 -0
- checkpoint-750/scheduler.bin +3 -0
- logs/dreambooth-flux2-klein-lora/1787132912.2150574/events.out.tfevents.1787132912.dc8010f0b725.2811.1 +3 -0
- logs/dreambooth-flux2-klein-lora/1787132912.2169197/hparams.yml +88 -0
- logs/dreambooth-flux2-klein-lora/events.out.tfevents.1787132912.dc8010f0b725.2811.0 +3 -0
- pytorch_lora_weights.safetensors +3 -0
README.md
CHANGED
|
@@ -1,3 +1,12 @@
|
|
| 1 |
---
|
| 2 |
license: cc-by-nc-nd-4.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
| 1 |
---
|
| 2 |
license: cc-by-nc-nd-4.0
|
| 3 |
+
library_name: diffusers
|
| 4 |
+
tags:
|
| 5 |
+
- text-to-image
|
| 6 |
+
- diffusers-training
|
| 7 |
+
- diffusers
|
| 8 |
+
- lora
|
| 9 |
+
- flux2-klein
|
| 10 |
+
- flux2-klein-diffusers
|
| 11 |
+
- template:sd-lora
|
| 12 |
---
|
checkpoint-1000/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:48a261a6d13b84ab95f1b854a37cfa18a795806ac8658e00c5a1ad4881ec1fb9
|
| 3 |
+
size 56634443
|
checkpoint-1000/pytorch_lora_weights.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:80fc70d7360e0288f7b7e4cee37de0fe71e6ef5e761ab53dccfc23c2eb2e561e
|
| 3 |
+
size 55599592
|
checkpoint-1000/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0b65bb649a505edd00f4343ecb956f93ee41f4d0ef51d9d5eff0c448ea49c6c3
|
| 3 |
+
size 14693
|
checkpoint-1000/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d5ca5fd1f3d50dfbd88c5f854e399c1c22a842d2231d41415da16bf924479e76
|
| 3 |
+
size 1401
|
checkpoint-1250/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b027d1752700776ea1f7910e6fd827bdcffd39e7266a2d9bf1acaf02f86c34f1
|
| 3 |
+
size 56634443
|
checkpoint-1250/pytorch_lora_weights.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d18429db399ddf2fddeab62bb5b1e004de366da02241d8c26cd6bc3636fadc67
|
| 3 |
+
size 55599592
|
checkpoint-1250/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:de4c242edac94a76415e47777feb4b5b496403666a702225d09a1f018ebf5b83
|
| 3 |
+
size 14693
|
checkpoint-1250/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:013ec8691bd290cdf8839f5567678211b588c8e265fe27bc6b8bb362844cfaf0
|
| 3 |
+
size 1401
|
checkpoint-250/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fbd73832f232ac570301d8504f68f240b646e8ea88b21fc48e0a9ad077ffe132
|
| 3 |
+
size 56634315
|
checkpoint-250/pytorch_lora_weights.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5719651233e58f710aa5472395b2b0142d81d953266d35297676bbecb133506e
|
| 3 |
+
size 55599592
|
checkpoint-250/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d6129f69ea3a80009287be59e2be072476bca87cd9b6259a662e95f58724a674
|
| 3 |
+
size 14693
|
checkpoint-250/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e9fd216eaa69ff4eb809e0f68c4cab14943314d6c9795506c007a05963b86fc8
|
| 3 |
+
size 1401
|
checkpoint-500/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cd79bdd0d7c659105511a385b326f3140a587e67a4485ec765c6c4039d2e9d8c
|
| 3 |
+
size 56634443
|
checkpoint-500/pytorch_lora_weights.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ec35791169afbd9adb9f53ed17871a33d2352d6727ac6355a59e97a9944e5f27
|
| 3 |
+
size 55599592
|
checkpoint-500/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:16ad70a5b3f8c8bb6024a8ff38fe0f3fb2e65107a0c9d69cb5605e55837e94c1
|
| 3 |
+
size 14693
|
checkpoint-500/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8baa45cc00e6da1eefbbbc2d50fb568432ef1e20241f0a3571b49dcb8d62f7f2
|
| 3 |
+
size 1401
|
checkpoint-750/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9388ab0dbb7c333f32c4bb9092a646a64681fb677684c7f9dfc1020e258a583d
|
| 3 |
+
size 56634443
|
checkpoint-750/pytorch_lora_weights.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d0a16776457b88dd4cb49d3259c87376573fea6b7518fb5ad8f82963334fc69d
|
| 3 |
+
size 55599592
|
checkpoint-750/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:74b6287dab829b25404f460a9c66361deba65a91e837e0a7389cd52ad555e757
|
| 3 |
+
size 14693
|
checkpoint-750/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:af27cfa62dd1a6d194c45c24a6db7d0c334459f3177682f8feacba9d4decd87b
|
| 3 |
+
size 1401
|
logs/dreambooth-flux2-klein-lora/1787132912.2150574/events.out.tfevents.1787132912.dc8010f0b725.2811.1
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8e9dabc7f3c2a6808cd99df342589c793d94a2b4d92dae764767d6cb97abaac7
|
| 3 |
+
size 4149
|
logs/dreambooth-flux2-klein-lora/1787132912.2169197/hparams.yml
ADDED
|
@@ -0,0 +1,88 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
adam_beta1: 0.9
|
| 2 |
+
adam_beta2: 0.999
|
| 3 |
+
adam_epsilon: 1.0e-08
|
| 4 |
+
adam_weight_decay: 0.0001
|
| 5 |
+
adam_weight_decay_text_encoder: 0.001
|
| 6 |
+
allow_tf32: false
|
| 7 |
+
aspect_ratio_buckets: null
|
| 8 |
+
bnb_quantization_config_path: null
|
| 9 |
+
cache_dir: null
|
| 10 |
+
cache_latents: true
|
| 11 |
+
caption_column: text
|
| 12 |
+
caption_dropout: 0.0
|
| 13 |
+
center_crop: false
|
| 14 |
+
checkpointing_steps: 250
|
| 15 |
+
checkpoints_total_limit: null
|
| 16 |
+
class_data_dir: null
|
| 17 |
+
class_prompt: null
|
| 18 |
+
dataloader_num_workers: 0
|
| 19 |
+
dataset_config_name: null
|
| 20 |
+
dataset_name: Retoe/shat3
|
| 21 |
+
do_fp8_training: false
|
| 22 |
+
enable_npu_flash_attention: false
|
| 23 |
+
final_validation_prompt: null
|
| 24 |
+
fsdp_text_encoder: false
|
| 25 |
+
gradient_accumulation_steps: 4
|
| 26 |
+
gradient_checkpointing: true
|
| 27 |
+
guidance_scale: 1.0
|
| 28 |
+
hub_model_id: Retoe/shatfine3
|
| 29 |
+
hub_token: null
|
| 30 |
+
image_column: image
|
| 31 |
+
instance_data_dir: null
|
| 32 |
+
instance_prompt: GROUNDZERO
|
| 33 |
+
learning_rate: 0.0001
|
| 34 |
+
local_rank: -1
|
| 35 |
+
logging_dir: logs
|
| 36 |
+
logit_mean: 0.0
|
| 37 |
+
logit_std: 1.0
|
| 38 |
+
lora_alpha: 16
|
| 39 |
+
lora_dropout: 0.0
|
| 40 |
+
lora_layers: null
|
| 41 |
+
lr_num_cycles: 1
|
| 42 |
+
lr_power: 1.0
|
| 43 |
+
lr_scheduler: constant
|
| 44 |
+
lr_warmup_steps: 100
|
| 45 |
+
max_grad_norm: 1.0
|
| 46 |
+
max_sequence_length: 512
|
| 47 |
+
max_train_steps: 1250
|
| 48 |
+
mixed_precision: bf16
|
| 49 |
+
mode_scale: 1.29
|
| 50 |
+
num_class_images: 100
|
| 51 |
+
num_train_epochs: 417
|
| 52 |
+
num_validation_images: 4
|
| 53 |
+
offload: true
|
| 54 |
+
optimizer: adamW
|
| 55 |
+
output_dir: shatfine-lora
|
| 56 |
+
pretrained_model_name_or_path: black-forest-labs/FLUX.2-klein-9B
|
| 57 |
+
prior_generation_precision: null
|
| 58 |
+
prior_loss_weight: 1.0
|
| 59 |
+
prodigy_beta3: null
|
| 60 |
+
prodigy_decouple: true
|
| 61 |
+
prodigy_safeguard_warmup: true
|
| 62 |
+
prodigy_use_bias_correction: true
|
| 63 |
+
push_to_hub: true
|
| 64 |
+
random_flip: false
|
| 65 |
+
rank: 16
|
| 66 |
+
repeats: 1
|
| 67 |
+
report_to: tensorboard
|
| 68 |
+
resolution: 1024
|
| 69 |
+
resume_from_checkpoint: null
|
| 70 |
+
revision: null
|
| 71 |
+
sample_batch_size: 4
|
| 72 |
+
scale_lr: false
|
| 73 |
+
seed: 0
|
| 74 |
+
skip_final_inference: false
|
| 75 |
+
text_encoder_lr: 5.0e-06
|
| 76 |
+
text_encoder_out_layers: '[10, 20, 30]'
|
| 77 |
+
train_batch_size: 1
|
| 78 |
+
upcast_before_saving: false
|
| 79 |
+
use_8bit_adam: true
|
| 80 |
+
use_aspect_ratio_buckets: false
|
| 81 |
+
validation_epochs: 25
|
| 82 |
+
validation_prompt: GROUNDZERO, a chibi deep sea diver character with a brass diving
|
| 83 |
+
helmet with a round front porthole, warm light on the face inside, a thick canvas
|
| 84 |
+
suit collar, and heavy bolted shoulder plates, character portrait bust, cropped
|
| 85 |
+
close-up, chibi 3D rendered toy figure style, black background
|
| 86 |
+
variant: null
|
| 87 |
+
weighting_scheme: none
|
| 88 |
+
with_prior_preservation: false
|
logs/dreambooth-flux2-klein-lora/events.out.tfevents.1787132912.dc8010f0b725.2811.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:964f936e1ea21aeed30374c712f3aba906f29a664cac3ea178fe08ced237214e
|
| 3 |
+
size 45654503
|
pytorch_lora_weights.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d18429db399ddf2fddeab62bb5b1e004de366da02241d8c26cd6bc3636fadc67
|
| 3 |
+
size 55599592
|