jpbalarini commited on
Commit
c50ae52
·
verified ·
1 Parent(s): 3b46a86

Upload 3 files

Browse files
config.yaml ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: scandinavian_flux_lora_v1
4
+ process:
5
+ - type: sd_trainer
6
+ training_folder: output
7
+ performance_log_every: 1000
8
+ device: cuda:0
9
+ network:
10
+ type: lora
11
+ linear: 32
12
+ linear_alpha: 32
13
+ save:
14
+ dtype: float16
15
+ save_every: 250
16
+ max_step_saves_to_keep: 4
17
+ datasets:
18
+ - folder_path: /home/jpb/Projects/ai-toolkit/input/scandinavian
19
+ caption_ext: txt
20
+ caption_dropout_rate: 0.05
21
+ shuffle_tokens: false
22
+ cache_latents_to_disk: true
23
+ resolution:
24
+ - 512
25
+ - 768
26
+ - 1024
27
+ train:
28
+ batch_size: 1
29
+ steps: 4000
30
+ gradient_accumulation_steps: 1
31
+ train_unet: true
32
+ train_text_encoder: false
33
+ gradient_checkpointing: true
34
+ noise_scheduler: flowmatch
35
+ optimizer: adamw8bit
36
+ lr: 0.0001
37
+ ema_config:
38
+ use_ema: true
39
+ ema_decay: 0.99
40
+ dtype: bf16
41
+ model:
42
+ name_or_path: black-forest-labs/FLUX.1-dev
43
+ is_flux: true
44
+ quantize: true
45
+ sample:
46
+ sampler: flowmatch
47
+ sample_every: 250
48
+ width: 1024
49
+ height: 1024
50
+ prompts:
51
+ - woman with red hair, playing chess at the park, bomb going off in the background
52
+ - a woman holding a coffee cup, in a beanie, sitting at a cafe
53
+ - a man showing off his cool new t shirt at the beach, a shark is jumping out
54
+ of the water in the background
55
+ - woman playing the guitar, on stage, singing a song, laser lights, punk rocker
56
+ - hipster man with a beard, building a chair, in a wood shop
57
+ - photo of a man, white background, medium shot, modeling clothing, studio lighting,
58
+ white backdrop
59
+ - a man holding a sign that says, 'this is a sign'
60
+ - a living room. Golden hour
61
+ - a kitchen. Golden hour
62
+ neg: ''
63
+ seed: 42
64
+ walk_seed: true
65
+ guidance_scale: 4
66
+ sample_steps: 20
67
+ meta:
68
+ name: scandinavian_flux_lora_v1
69
+ version: '1.0'
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb240aee70a641e4e97fadc94cdeb500d8bda91b2474c691e85504d759f1b70d
3
+ size 345500484
scandinavian_flux_lora_v1.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3fbb11d023d1aa19c616daec2e0f85464f55fa81e5440cceadedaeca06f60aaf
3
+ size 343805416