pranaykoppula
/

rdjml

Model card Files Files and versions

pranaykoppula commited on Oct 25, 2023

Commit

cf1dcfe

·

1 Parent(s): f207030

feat: upload rdjml checkpoint

Files changed (2) hide show

rdjml_config/config_file.toml +1 -1
rdjml_config/dataset_config.toml +1 -1

rdjml_config/config_file.toml CHANGED Viewed

@@ -1,7 +1,7 @@
 [model_arguments]
 v2 = false
 v_parameterization = false
-pretrained_model_name_or_path = "/content/pretrained_model/OpenJourney-v4.ckpt"
 [optimizer_arguments]
 min_snr_gamma = 5

 [model_arguments]
 v2 = false
 v_parameterization = false
+pretrained_model_name_or_path = "/content/pretrained_model/rdjml.ckpt"
 [optimizer_arguments]
 min_snr_gamma = 5

rdjml_config/dataset_config.toml CHANGED Viewed

@@ -9,7 +9,7 @@ flip_aug = false
 color_aug = false
 [[datasets.subsets]]
 image_dir = "/content/dreambooth/train_data"
-class_tokens = "robert-downey"
 num_repeats = 10

 color_aug = false
 [[datasets.subsets]]
 image_dir = "/content/dreambooth/train_data"
+class_tokens = "checkshirt"
 num_repeats = 10