TheAwakenOne commited on
Commit
1dde2c9
1 Parent(s): 59ce361

Upload folder using huggingface_hub

Browse files
Files changed (44) hide show
  1. README.md +48 -0
  2. dataset.toml +14 -0
  3. sample/watercolor_000100_00_20241009225031.png +0 -0
  4. sample/watercolor_000100_01_20241009225248.png +0 -0
  5. sample/watercolor_000100_02_20241009225501.png +0 -0
  6. sample/watercolor_000100_03_20241009225717.png +0 -0
  7. sample/watercolor_000200_00_20241009231519.png +0 -0
  8. sample/watercolor_000200_01_20241009231736.png +0 -0
  9. sample/watercolor_000200_02_20241009231953.png +0 -0
  10. sample/watercolor_000200_03_20241009232210.png +0 -0
  11. sample/watercolor_000300_00_20241009234003.png +0 -0
  12. sample/watercolor_000300_01_20241009234217.png +0 -0
  13. sample/watercolor_000300_02_20241009234432.png +0 -0
  14. sample/watercolor_000300_03_20241009234645.png +0 -0
  15. sample/watercolor_000400_00_20241010000433.png +0 -0
  16. sample/watercolor_000400_01_20241010000648.png +0 -0
  17. sample/watercolor_000400_02_20241010000904.png +0 -0
  18. sample/watercolor_000400_03_20241010001116.png +0 -0
  19. sample/watercolor_000500_00_20241010002904.png +0 -0
  20. sample/watercolor_000500_01_20241010003117.png +0 -0
  21. sample/watercolor_000500_02_20241010003332.png +0 -0
  22. sample/watercolor_000500_03_20241010003545.png +0 -0
  23. sample/watercolor_000600_00_20241010005333.png +0 -0
  24. sample/watercolor_000600_01_20241010005545.png +0 -0
  25. sample/watercolor_000600_02_20241010005800.png +0 -0
  26. sample/watercolor_000600_03_20241010010012.png +0 -0
  27. sample/watercolor_000700_00_20241010011801.png +0 -0
  28. sample/watercolor_000700_01_20241010012016.png +0 -0
  29. sample/watercolor_000700_02_20241010012230.png +0 -0
  30. sample/watercolor_000700_03_20241010012440.png +0 -0
  31. sample/watercolor_000800_00_20241010014233.png +0 -0
  32. sample/watercolor_000800_01_20241010014443.png +0 -0
  33. sample/watercolor_000800_02_20241010014656.png +0 -0
  34. sample/watercolor_000800_03_20241010014911.png +0 -0
  35. sample/watercolor_000900_00_20241010020653.png +0 -0
  36. sample/watercolor_000900_01_20241010020907.png +0 -0
  37. sample/watercolor_000900_02_20241010021119.png +0 -0
  38. sample/watercolor_000900_03_20241010021332.png +0 -0
  39. sample_prompts.txt +4 -0
  40. train.bat +39 -0
  41. watercolor-000004.safetensors +3 -0
  42. watercolor-000008.safetensors +3 -0
  43. watercolor-000012.safetensors +3 -0
  44. watercolor.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - text-to-image
4
+ - flux
5
+ - lora
6
+ - diffusers
7
+ - template:sd-lora
8
+ - fluxgym
9
+ widget:
10
+ - output:
11
+ url: sample/watercolor_000900_00_20241010020653.png
12
+ text: WAT3R a watercolor painting of a man wearing a purple neck tie against a
13
+ white background. Her face is clearly visible, with her eyes looking directly
14
+ at the viewer and her lips slightly parted
15
+ - output:
16
+ url: sample/watercolor_000900_01_20241010020907.png
17
+ text: WAT3R a vibrant watercolor butterfly tattoo design on a white background.
18
+ The butterfly is composed of a variety of colors
19
+ - output:
20
+ url: sample/watercolor_000900_02_20241010021119.png
21
+ text: WAT3R a beautiful watercolor painting of a woman with blue hair against
22
+ a white background
23
+ - output:
24
+ url: sample/watercolor_000900_03_20241010021332.png
25
+ text: WAT3R a beautiful watercolor painting of a horse with a white background.
26
+ The horse is painted in a variety of colors, including blues, greens, yellows,
27
+ and reds, creating a vibrant and eye-catching piece of art
28
+ base_model: black-forest-labs/FLUX.1-dev
29
+ instance_prompt: WAT3R
30
+ license: other
31
+ license_name: flux-1-dev-non-commercial-license
32
+ license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
33
+ ---
34
+
35
+ # WaterColor
36
+
37
+ A Flux LoRA trained on a local computer with [Fluxgym](https://github.com/cocktailpeanut/fluxgym)
38
+
39
+ <Gallery />
40
+
41
+ ## Trigger words
42
+
43
+ You should use `WAT3R` to trigger the image generation.
44
+
45
+ ## Download model and use it with ComfyUI, AUTOMATIC1111, SD.Next, Invoke AI, Forge, etc.
46
+
47
+ Weights for this model are available in Safetensors format.
48
+
dataset.toml ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [general]
2
+ shuffle_caption = false
3
+ caption_extension = '.txt'
4
+ keep_tokens = 1
5
+
6
+ [[datasets]]
7
+ resolution = 512
8
+ batch_size = 1
9
+ keep_tokens = 1
10
+
11
+ [[datasets.subsets]]
12
+ image_dir = 'E:\AIOne\pinokio\api\fluxgym.git\datasets\watercolor'
13
+ class_tokens = 'WAT3R'
14
+ num_repeats = 10
sample/watercolor_000100_00_20241009225031.png ADDED
sample/watercolor_000100_01_20241009225248.png ADDED
sample/watercolor_000100_02_20241009225501.png ADDED
sample/watercolor_000100_03_20241009225717.png ADDED
sample/watercolor_000200_00_20241009231519.png ADDED
sample/watercolor_000200_01_20241009231736.png ADDED
sample/watercolor_000200_02_20241009231953.png ADDED
sample/watercolor_000200_03_20241009232210.png ADDED
sample/watercolor_000300_00_20241009234003.png ADDED
sample/watercolor_000300_01_20241009234217.png ADDED
sample/watercolor_000300_02_20241009234432.png ADDED
sample/watercolor_000300_03_20241009234645.png ADDED
sample/watercolor_000400_00_20241010000433.png ADDED
sample/watercolor_000400_01_20241010000648.png ADDED
sample/watercolor_000400_02_20241010000904.png ADDED
sample/watercolor_000400_03_20241010001116.png ADDED
sample/watercolor_000500_00_20241010002904.png ADDED
sample/watercolor_000500_01_20241010003117.png ADDED
sample/watercolor_000500_02_20241010003332.png ADDED
sample/watercolor_000500_03_20241010003545.png ADDED
sample/watercolor_000600_00_20241010005333.png ADDED
sample/watercolor_000600_01_20241010005545.png ADDED
sample/watercolor_000600_02_20241010005800.png ADDED
sample/watercolor_000600_03_20241010010012.png ADDED
sample/watercolor_000700_00_20241010011801.png ADDED
sample/watercolor_000700_01_20241010012016.png ADDED
sample/watercolor_000700_02_20241010012230.png ADDED
sample/watercolor_000700_03_20241010012440.png ADDED
sample/watercolor_000800_00_20241010014233.png ADDED
sample/watercolor_000800_01_20241010014443.png ADDED
sample/watercolor_000800_02_20241010014656.png ADDED
sample/watercolor_000800_03_20241010014911.png ADDED
sample/watercolor_000900_00_20241010020653.png ADDED
sample/watercolor_000900_01_20241010020907.png ADDED
sample/watercolor_000900_02_20241010021119.png ADDED
sample/watercolor_000900_03_20241010021332.png ADDED
sample_prompts.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ WAT3R a watercolor painting of a man wearing a purple neck tie against a white background. Her face is clearly visible, with her eyes looking directly at the viewer and her lips slightly parted
2
+ WAT3R a vibrant watercolor butterfly tattoo design on a white background. The butterfly is composed of a variety of colors
3
+ WAT3R a beautiful watercolor painting of a woman with blue hair against a white background
4
+ WAT3R a beautiful watercolor painting of a horse with a white background. The horse is painted in a variety of colors, including blues, greens, yellows, and reds, creating a vibrant and eye-catching piece of art
train.bat ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ accelerate launch ^
2
+ --mixed_precision bf16 ^
3
+ --num_cpu_threads_per_process 1 ^
4
+ sd-scripts/flux_train_network.py ^
5
+ --pretrained_model_name_or_path "E:\AIOne\pinokio\api\fluxgym.git\models\unet\bdsqlsz\flux1-dev2pro-single\flux1-dev2pro.safetensors" ^
6
+ --clip_l "E:\AIOne\pinokio\api\fluxgym.git\models\clip\clip_l.safetensors" ^
7
+ --t5xxl "E:\AIOne\pinokio\api\fluxgym.git\models\clip\t5xxl_fp16.safetensors" ^
8
+ --ae "E:\AIOne\pinokio\api\fluxgym.git\models\vae\ae.sft" ^
9
+ --cache_latents_to_disk ^
10
+ --save_model_as safetensors ^
11
+ --sdpa --persistent_data_loader_workers ^
12
+ --max_data_loader_n_workers 2 ^
13
+ --seed 42 ^
14
+ --gradient_checkpointing ^
15
+ --mixed_precision bf16 ^
16
+ --save_precision bf16 ^
17
+ --network_module networks.lora_flux ^
18
+ --network_dim 4 ^
19
+ --optimizer_type adafactor ^
20
+ --optimizer_args "relative_step=False" "scale_parameter=False" "warmup_init=False" ^
21
+ --split_mode ^
22
+ --network_args "train_blocks=single" ^
23
+ --lr_scheduler constant_with_warmup ^
24
+ --max_grad_norm 0.0 ^--sample_prompts="E:\AIOne\pinokio\api\fluxgym.git\outputs\watercolor\sample_prompts.txt" --sample_every_n_steps="100" ^
25
+ --learning_rate 8e-4 ^
26
+ --cache_text_encoder_outputs ^
27
+ --cache_text_encoder_outputs_to_disk ^
28
+ --fp8_base ^
29
+ --highvram ^
30
+ --max_train_epochs 16 ^
31
+ --save_every_n_epochs 4 ^
32
+ --dataset_config "E:\AIOne\pinokio\api\fluxgym.git\outputs\watercolor\dataset.toml" ^
33
+ --output_dir "E:\AIOne\pinokio\api\fluxgym.git\outputs\watercolor" ^
34
+ --output_name watercolor ^
35
+ --timestep_sampling shift ^
36
+ --discrete_flow_shift 3.1582 ^
37
+ --model_prediction_type raw ^
38
+ --guidance_scale 1 ^
39
+ --loss_type l2 ^
watercolor-000004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3715d36bcbd3d18befdb0dcfea13b66725bd9a330ffde218c120f09b1981912
3
+ size 18211812
watercolor-000008.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1eb1cea1dabed5053e49785effc2fd80f24843e65f08d94534822f5386e848e1
3
+ size 18211812
watercolor-000012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10f42e9bf534f9dfae1228c6bfbfd8fa2071d5fb82728ac52826718f3cc1d09b
3
+ size 18211812
watercolor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e2e5c99f7023383c9ed0fc2f0582536e95c6873b95e5c6fc84f191d49a9c1d3
3
+ size 18211812