Commit ·
e594d54
1
Parent(s): 185021d
feat: upload reaimer-v0-e5-5 checkpoint
Browse files
v0_config/config_file.toml
CHANGED
|
@@ -14,7 +14,7 @@ lr_warmup_steps = 0
|
|
| 14 |
|
| 15 |
[dataset_arguments]
|
| 16 |
debug_dataset = false
|
| 17 |
-
in_json = "/content/fine_tune/meta_lat.json"
|
| 18 |
train_data_dir = "/content/fine_tune/train_data"
|
| 19 |
dataset_repeats = 1
|
| 20 |
shuffle_caption = true
|
|
@@ -29,7 +29,7 @@ token_warmup_step = 0
|
|
| 29 |
|
| 30 |
[training_arguments]
|
| 31 |
output_dir = "/content/fine_tune/output"
|
| 32 |
-
output_name = "reaimer-v0-
|
| 33 |
save_precision = "fp16"
|
| 34 |
save_n_epoch_ratio = 1
|
| 35 |
save_state = true
|
|
@@ -45,7 +45,7 @@ gradient_checkpointing = false
|
|
| 45 |
gradient_accumulation_steps = 1
|
| 46 |
mixed_precision = "fp16"
|
| 47 |
logging_dir = "/content/fine_tune/logs"
|
| 48 |
-
log_prefix = "reaimer-v0-
|
| 49 |
noise_offset = 0.12
|
| 50 |
|
| 51 |
[sample_prompt_arguments]
|
|
|
|
| 14 |
|
| 15 |
[dataset_arguments]
|
| 16 |
debug_dataset = false
|
| 17 |
+
in_json = "/content/fine_tune/config/meta_lat.json"
|
| 18 |
train_data_dir = "/content/fine_tune/train_data"
|
| 19 |
dataset_repeats = 1
|
| 20 |
shuffle_caption = true
|
|
|
|
| 29 |
|
| 30 |
[training_arguments]
|
| 31 |
output_dir = "/content/fine_tune/output"
|
| 32 |
+
output_name = "reaimer-v0-5-5"
|
| 33 |
save_precision = "fp16"
|
| 34 |
save_n_epoch_ratio = 1
|
| 35 |
save_state = true
|
|
|
|
| 45 |
gradient_accumulation_steps = 1
|
| 46 |
mixed_precision = "fp16"
|
| 47 |
logging_dir = "/content/fine_tune/logs"
|
| 48 |
+
log_prefix = "reaimer-v0-e5-5"
|
| 49 |
noise_offset = 0.12
|
| 50 |
|
| 51 |
[sample_prompt_arguments]
|