pranaykoppula commited on
Commit
2a3ea73
·
1 Parent(s): 398955c

feat: upload winsletml checkpoint

Browse files
winsletml_config/config_file.toml CHANGED
@@ -1,7 +1,7 @@
1
  [model_arguments]
2
  v2 = false
3
  v_parameterization = false
4
- pretrained_model_name_or_path = "/content/pretrained_model/winsletml.ckpt"
5
  vae = "/content/vae/stablediffusion.vae.pt"
6
 
7
  [optimizer_arguments]
 
1
  [model_arguments]
2
  v2 = false
3
  v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/Anything-v3-1.safetensors"
5
  vae = "/content/vae/stablediffusion.vae.pt"
6
 
7
  [optimizer_arguments]
winsletml_config/dataset_config.toml CHANGED
@@ -9,7 +9,7 @@ flip_aug = false
9
  color_aug = false
10
  [[datasets.subsets]]
11
  image_dir = "/content/dreambooth/train_data"
12
- class_tokens = "eiffel-tower"
13
  num_repeats = 10
14
 
15
 
 
9
  color_aug = false
10
  [[datasets.subsets]]
11
  image_dir = "/content/dreambooth/train_data"
12
+ class_tokens = "kate-winslet"
13
  num_repeats = 10
14
 
15