feat: upload last checkpoint

config_file.toml

@@ -2,6 +2,7 @@
 v2 = false
 v_parameterization = false
 pretrained_model_name_or_path = "/content/pretrained_model/Stable-Diffusion-v1-5.safetensors"
+vae = "/content/vae/stablediffusion.vae.pt"
 
 [optimizer_arguments]
 min_snr_gamma = 5

dataset_config.toml

@@ -9,15 +9,9 @@ flip_aug = false
 color_aug = false
 [[datasets.subsets]]
 image_dir = "/content/dreambooth/train_data"
-class_tokens = "hugh jackman"
+class_tokens = "hugh-jackman"
 num_repeats = 10
 
-[[datasets.subsets]]
-is_reg = true
-image_dir = "/content/dreambooth/reg_data"
-class_tokens = "hugh jackman"
-num_repeats = 1
-
 
 [general]
 enable_bucket = true