Upload folder using huggingface_hub

README.md

@@ -0,0 +1,31 @@
+---
+tags:
+- text-to-image
+- flux
+- lora
+- diffusers
+- template:sd-lora
+- fluxgym
+
+
+base_model: black-forest-labs/FLUX.1-dev
+instance_prompt: tok_man
+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
+---
+
+# tok
+
+A Flux LoRA trained on a local computer with [Fluxgym](https://github.com/cocktailpeanut/fluxgym)
+
+<Gallery />
+
+## Trigger words
+
+You should use `tok_man` 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

@@ -4,11 +4,11 @@ caption_extension = '.txt'
 keep_tokens = 1
 
 [[datasets]]
-resolution = 512
+resolution = 1024
 batch_size = 1
 keep_tokens = 1
 
   [[datasets.subsets]]
   image_dir = '/root/fluxgym/datasets/tok'
-  class_tokens = 'ohwx-man'
-  num_repeats = 8
+  class_tokens = 'tok_man'
+  num_repeats = 11

sample_prompts.txt

@@ -1 +1,3 @@
-ohwx-man
+tok_man
+tok a man
+photo tok

train.sh

@@ -22,7 +22,7 @@ env/bin/accelerate launch \
   --cache_text_encoder_outputs_to_disk \
   --fp8_base \
   --highvram \
-  --max_train_epochs 12 \
+  --max_train_epochs 10 \
   --save_every_n_epochs 4 \
   --dataset_config "/root/fluxgym/outputs/tok/dataset.toml" \
   --output_dir "/root/fluxgym/outputs/tok" \