Alterneko commited on
Commit
b3fbb17
1 Parent(s): 960d626

feat: upload Matsuriv1-140k-steps checkpoint

Browse files
Files changed (1) hide show
  1. v1_config/config_file.toml +5 -4
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-120k-steps.ckpt"
 
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-130k-steps"
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-130k-steps"
49
 
50
  [sample_prompt_arguments]
51
- sample_every_n_steps = 250
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]