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