Upload folder using huggingface_hub

README.md

@@ -0,0 +1,42 @@
+---
+tags:
+- text-to-image
+- flux
+- lora
+- diffusers
+- template:sd-lora
+- fluxgym
+widget:
+-   output:
+        url: sample/optimus_001000_00_20240916125554.png
+    text: optimus is cleaning the house with broomstick
+-   output:
+        url: sample/optimus_001000_01_20240916125612.png
+    text: optimus is competing in a bboy break dancing competition
+-   output:
+        url: sample/optimus_001000_02_20240916125629.png
+    text: optimus is playing the game of go, against a human
+-   output:
+        url: sample/optimus_001000_03_20240916125646.png
+    text: optimus is driving a car
+base_model: black-forest-labs/FLUX.1-dev
+instance_prompt: optimus
+license: other
+license_name: flux-1-dev-non-commercial-license
+license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
+---
+
+# optimus
+
+Trained with [Fluxgym](https://github.com/cocktailpeanut/fluxgym)
+
+<Gallery />
+
+## Trigger words
+
+You should use `optimus` to trigger the image generation.
+
+## Download model and use it with ComfyUI, AUTOMATIC1111, SD.Next, Invoke AI, Forge, etc.
+
+Weights for this model are available in Safetensors format.
+

dataset.toml

@@ -0,0 +1,14 @@
+[general]
+shuffle_caption = false
+caption_extension = '.txt'
+keep_tokens = 1
+
+[[datasets]]
+resolution = 512
+batch_size = 1
+keep_tokens = 1
+
+  [[datasets.subsets]]
+  image_dir = 'C:\pinokio\api\fluxgym.git\datasets\optimus'
+  class_tokens = 'optimus'
+  num_repeats = 8

optimus-000004.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:28438db28b90c4b8a5ad10b73df616ba9395e7c2d7a7367d6f7d29343bc1764d
+size 39757112

optimus-000008.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1ff3fedc9d98d5b82208b3e3046baf89cd76796f95138b8f57a9ff9617113e3c
+size 39757112

optimus-000012.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f1890035daf2b3b8875ee1e2cdc59227d6207b9f87cc7c48470744b2efdfac5f
+size 39757112

optimus.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9cac6a9210e7a706e642863eb96eca4f68c53770889798a2c4e38d0fc27279c4
+size 39757112

sample_prompts.txt

@@ -0,0 +1,4 @@
+optimus is cleaning the house with broomstick
+optimus is competing in a bboy break dancing competition
+optimus is playing the game of go, against a human
+optimus is driving a car

train.bat

@@ -0,0 +1,34 @@
+accelerate launch ^
+  --mixed_precision bf16 ^
+  --num_cpu_threads_per_process 1 ^
+  sd-scripts/flux_train_network.py ^
+  --pretrained_model_name_or_path "C:\pinokio\api\fluxgym.git\models\unet\flux1-dev.sft" ^
+  --clip_l "C:\pinokio\api\fluxgym.git\models\clip\clip_l.safetensors" ^
+  --t5xxl "C:\pinokio\api\fluxgym.git\models\clip\t5xxl_fp16.safetensors" ^
+  --ae "C:\pinokio\api\fluxgym.git\models\vae\ae.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 ^--sample_prompts="C:\pinokio\api\fluxgym.git\outputs\optimus\sample_prompts.txt" --sample_every_n_steps="200" ^
+  --learning_rate 8e-4 ^
+  --cache_text_encoder_outputs ^
+  --cache_text_encoder_outputs_to_disk ^
+  --fp8_base ^
+  --highvram ^
+  --max_train_epochs 16 ^
+  --save_every_n_epochs 4 ^
+  --dataset_config "C:\pinokio\api\fluxgym.git\outputs\optimus\dataset.toml" ^
+  --output_dir "C:\pinokio\api\fluxgym.git\outputs\optimus" ^
+  --output_name optimus ^
+  --timestep_sampling shift ^
+  --discrete_flow_shift 3.1582 ^
+  --model_prediction_type raw ^
+  --guidance_scale 1 ^
+  --loss_type l2 ^