Dangekos commited on
Commit
2118522
·
verified ·
1 Parent(s): 6fb7878

feat: upload model

Browse files
epoch5/anima_edit_1ref_1024.toml ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ output_dir = '/content/anima_training'
2
+
3
+ dataset = '/content/anima_edit_dataset.toml'
4
+
5
+ epochs = 100
6
+
7
+ micro_batch_size_per_gpu = 8
8
+
9
+ pipeline_stages = 1
10
+
11
+ gradient_accumulation_steps = 1
12
+
13
+ gradient_clipping = 1.0
14
+
15
+ warmup_steps = 100
16
+
17
+ eval_every_n_epochs = 1
18
+ eval_before_first_step = false
19
+ eval_micro_batch_size_per_gpu = 1
20
+ eval_gradient_accumulation_steps = 1
21
+
22
+ save_every_n_epochs = 1
23
+
24
+ checkpoint_every_n_epochs = 1
25
+
26
+ activation_checkpointing = true
27
+
28
+ partition_method = 'parameters'
29
+
30
+ save_dtype = 'bfloat16'
31
+
32
+ caching_batch_size = 1
33
+
34
+ steps_per_print = 5
35
+
36
+ [model]
37
+ type = 'anima'
38
+
39
+ transformer_path = '/content/models/epoch2/model.safetensors'
40
+
41
+ llm_path = '/content/models/split_files/text_encoders/qwen_3_06b_base.safetensors'
42
+
43
+ vae_path = '/content/models/split_files/vae/qwen_image_vae.safetensors'
44
+
45
+ dtype = 'bfloat16'
46
+
47
+ llm_adapter_lr = 0
48
+
49
+ [optimizer]
50
+ type = 'adamw_optimi'
51
+ lr = 3e-6
52
+ betas = [0.9, 0.99]
53
+ weight_decay = 0.01
54
+ eps = 1e-8
55
+
56
+ [monitoring]
57
+ enable_wandb = true
58
+
59
+ wandb_api_key = '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'
60
+
61
+ wandb_tracker_name = 'anima_edit_training'
62
+
63
+ wandb_run_name = '3e-6_batch8_runv1_2'
epoch5/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4fda4452075a37828d565d2798f3181153caf4c77f21f84c64ec587560946e3
3
+ size 4182243184