yachty66 commited on
Commit
702c9dd
1 Parent(s): a250f8d

Add model files

Browse files
Files changed (1) hide show
  1. config.yaml +72 -0
config.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: my_second_flux_lora_v1
4
+ process:
5
+ - type: sd_trainer
6
+ training_folder: output2
7
+ device: cuda:0
8
+ network:
9
+ type: lora
10
+ linear: 16
11
+ linear_alpha: 16
12
+ save:
13
+ dtype: float16
14
+ save_every: 250
15
+ max_step_saves_to_keep: 4
16
+ push_to_hub: true
17
+ datasets:
18
+ - folder_path: input2/
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: 2000
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 horse is a DJ at a night club, fish eye lens, smoke machine, lazer lights,
54
+ holding a martini
55
+ - a man showing off his cool new t shirt at the beach, a shark is jumping out
56
+ of the water in the background
57
+ - a bear building a log cabin in the snow covered mountains
58
+ - woman playing the guitar, on stage, singing a song, laser lights, punk rocker
59
+ - hipster man with a beard, building a chair, in a wood shop
60
+ - photo of a man, white background, medium shot, modeling clothing, studio lighting,
61
+ white backdrop
62
+ - a man holding a sign that says, 'this is a sign'
63
+ - a bulldog, in a post apocalyptic world, with a shotgun, in a leather jacket,
64
+ in a desert, with a motorcycle
65
+ neg: ''
66
+ seed: 42
67
+ walk_seed: true
68
+ guidance_scale: 4
69
+ sample_steps: 20
70
+ meta:
71
+ name: my_second_flux_lora_v1
72
+ version: '1.0'