armhebb commited on
Commit
5278b66
·
verified ·
1 Parent(s): 212db77

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. config.yaml +48 -0
  2. flat-model-3.safetensors +3 -0
  3. optimizer.pt +3 -0
config.yaml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: flat-model-3
4
+ process:
5
+ - type: sd_trainer
6
+ training_folder: ./artifacts/weights
7
+ device: cuda:0
8
+ trigger_word: clean lines flat fashion technical sketch in the style of TOK
9
+ network:
10
+ type: lora
11
+ linear: 32
12
+ linear_alpha: 32
13
+ save:
14
+ dtype: float16
15
+ save_every: 3000
16
+ max_step_saves_to_keep: 4
17
+ datasets:
18
+ - folder_path: ./artifacts/dataset/flat-model-3
19
+ caption_ext: txt
20
+ caption_dropout_rate: 0.1
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: 3000
30
+ gradient_accumulation_steps: 1
31
+ train_unet: true
32
+ train_text_encoder: false
33
+ disable_sampling: true
34
+ gradient_checkpointing: true
35
+ noise_scheduler: flowmatch
36
+ optimizer: adamw8bit
37
+ lr: 0.0001
38
+ ema_config:
39
+ use_ema: true
40
+ ema_decay: 0.99
41
+ dtype: bf16
42
+ model:
43
+ name_or_path: ./artifacts/flux_base
44
+ is_flux: true
45
+ quantize: true
46
+ meta:
47
+ name: flat-model-3
48
+ version: '1.0'
flat-model-3.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f9f417f36b23d56ac24504c473ee8735ae77dd47078cb694b2c28fb24cd15e40
3
+ size 343805552
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1d11cb9b7d2fffb033fbf1050e262cd16781e5d55f62d1595128262e3fbc270
3
+ size 350203012