feat: upload Matsuriv1-140k-steps checkpoint
Browse files
v1_config/config_file.toml
CHANGED
@@ -1,7 +1,8 @@
|
|
1 |
[model_arguments]
|
2 |
v2 = false
|
3 |
v_parameterization = false
|
4 |
-
pretrained_model_name_or_path = "/content/pretrained_model/Matsuriv1-
|
|
|
5 |
|
6 |
[optimizer_arguments]
|
7 |
min_snr_gamma = 5
|
@@ -29,7 +30,7 @@ token_warmup_step = 0
|
|
29 |
|
30 |
[training_arguments]
|
31 |
output_dir = "/content/fine_tune/output"
|
32 |
-
output_name = "Matsuriv1-
|
33 |
save_precision = "fp16"
|
34 |
save_every_n_steps = 12131
|
35 |
save_state = true
|
@@ -45,10 +46,10 @@ gradient_accumulation_steps = 1
|
|
45 |
mixed_precision = "fp16"
|
46 |
clip_skip = 2
|
47 |
logging_dir = "/content/fine_tune/logs"
|
48 |
-
log_prefix = "Matsuriv1-
|
49 |
|
50 |
[sample_prompt_arguments]
|
51 |
-
sample_every_n_steps =
|
52 |
sample_sampler = "ddim"
|
53 |
|
54 |
[saving_arguments]
|
|
|
1 |
[model_arguments]
|
2 |
v2 = false
|
3 |
v_parameterization = false
|
4 |
+
pretrained_model_name_or_path = "/content/pretrained_model/Matsuriv1-130k-steps.ckpt"
|
5 |
+
resume = "/content/fine_tune/state"
|
6 |
|
7 |
[optimizer_arguments]
|
8 |
min_snr_gamma = 5
|
|
|
30 |
|
31 |
[training_arguments]
|
32 |
output_dir = "/content/fine_tune/output"
|
33 |
+
output_name = "Matsuriv1-140k-steps"
|
34 |
save_precision = "fp16"
|
35 |
save_every_n_steps = 12131
|
36 |
save_state = true
|
|
|
46 |
mixed_precision = "fp16"
|
47 |
clip_skip = 2
|
48 |
logging_dir = "/content/fine_tune/logs"
|
49 |
+
log_prefix = "Matsuriv1-140k-steps"
|
50 |
|
51 |
[sample_prompt_arguments]
|
52 |
+
sample_every_n_steps = 500
|
53 |
sample_sampler = "ddim"
|
54 |
|
55 |
[saving_arguments]
|