Upload folder using huggingface_hub

README.md

@@ -9,7 +9,7 @@ tags:
 
 
 base_model: black-forest-labs/FLUX.1-dev
-instance_prompt: T0M
+instance_prompt: mr-tom
 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
@@ -23,7 +23,7 @@ A Flux LoRA trained on a local computer with [Fluxgym](https://github.com/cockta
 
 ## Trigger words
 
-You should use `T0M` to trigger the image generation.
+You should use `mr-tom` to trigger the image generation.
 
 ## Download model and use it with ComfyUI, AUTOMATIC1111, SD.Next, Invoke AI, Forge, etc.
 

dataset.toml

@@ -10,5 +10,5 @@ keep_tokens = 1
 
   [[datasets.subsets]]
   image_dir = '/app/fluxgym/datasets/tom'
-  class_tokens = 'T0M'
-  num_repeats = 2
+  class_tokens = 'mr-tom'
+  num_repeats = 10

sample_prompts.txt

@@ -1 +1 @@
-T0M
+mr-tom

tom-000004.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:842c3bc95f8cb3f1fbddb1b29553f05e374a5c350ec98e0a2cd41678f8671e6a
-size 39760784
+oid sha256:0daf663492a5dcafad9d1df665d6fcc77687cf94cf4fd9261925a127fd8daae1
+size 39766344

tom-000008.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e6497fd80a738bb693e8b6f251decbbb57c08544cd55823b5e6de715992d7aad
-size 39760784
+oid sha256:de2ff6519654f7d23ad170c2a3f42d01ded2d73f7837d7c152277855379468cb
+size 39766344

tom-000012.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cf4d2a62633ef42b3ec8eeaa9036a1c2789f7a6f2fe9389c126a68a617eeaaa5
-size 39760784
+oid sha256:b72b54bc7e56a64e35a2bb5b414da4a077779404fa4ad5fe87eea391a6bebfef
+size 39766344

tom.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:46e303d7b6b11d6a8d13f6f1f06675511e1a2380672550e43af7805a0e79d956
-size 39760784
+oid sha256:38ca2603d4f9918bc1c9d5763e1ff9447e37e6381c6899e12cb4803a78d84f77
+size 39766344

train.sh

@@ -31,4 +31,8 @@ accelerate launch \
   --discrete_flow_shift 3.1582 \
   --model_prediction_type raw \
   --guidance_scale 1 \
-  --loss_type l2 \
+  --loss_type l2 \
+  --alpha_mask \
+  --apply_t5_attn_mask \
+  --full_bf16 \
+  --masked_loss