pranaykoppula commited on
Commit
625fe37
1 Parent(s): 4a472e0

feat: upload last checkpoint

Browse files
Files changed (3) hide show
  1. config_file.toml +48 -0
  2. dataset_config.toml +22 -0
  3. sample_prompt.txt +2 -0
config_file.toml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/Stable-Diffusion-v1-5.safetensors"
5
+ vae = "/content/vae/stablediffusion.vae.pt"
6
+
7
+ [optimizer_arguments]
8
+ min_snr_gamma = 5
9
+ optimizer_type = "AdamW8bit"
10
+ learning_rate = 2e-6
11
+ max_grad_norm = 1.0
12
+ lr_scheduler = "constant"
13
+ lr_warmup_steps = 0
14
+
15
+ [dataset_arguments]
16
+ cache_latents = true
17
+ debug_dataset = false
18
+ vae_batch_size = 1
19
+
20
+ [training_arguments]
21
+ output_dir = "/content/drive/MyDrive/dreambooth/output"
22
+ output_name = "last"
23
+ save_precision = "fp16"
24
+ save_n_epoch_ratio = 1
25
+ save_state = false
26
+ train_batch_size = 4
27
+ max_token_length = 225
28
+ mem_eff_attn = false
29
+ xformers = true
30
+ max_train_steps = 2500
31
+ max_data_loader_n_workers = 8
32
+ persistent_data_loader_workers = true
33
+ gradient_checkpointing = false
34
+ gradient_accumulation_steps = 1
35
+ mixed_precision = "fp16"
36
+ clip_skip = 2
37
+ logging_dir = "/content/dreambooth/logs"
38
+ log_prefix = "last"
39
+
40
+ [sample_prompt_arguments]
41
+ sample_every_n_steps = 999999
42
+ sample_sampler = "ddim"
43
+
44
+ [dreambooth_arguments]
45
+ prior_loss_weight = 1.0
46
+
47
+ [saving_arguments]
48
+ save_model_as = "ckpt"
dataset_config.toml ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [[datasets]]
2
+ resolution = 512
3
+ min_bucket_reso = 256
4
+ max_bucket_reso = 1024
5
+ caption_dropout_rate = 0
6
+ caption_tag_dropout_rate = 0
7
+ caption_dropout_every_n_epochs = 0
8
+ flip_aug = false
9
+ color_aug = false
10
+ [[datasets.subsets]]
11
+ image_dir = "/content/dreambooth/train_data"
12
+ class_tokens = "hugh-jackman"
13
+ num_repeats = 10
14
+
15
+
16
+ [general]
17
+ enable_bucket = true
18
+ caption_extension = ".txt"
19
+ shuffle_caption = true
20
+ keep_tokens = 0
21
+ bucket_reso_steps = 64
22
+ bucket_no_upscale = false
sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ masterpiece, best quality, 1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt --n lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry --w 512 --h 768 --l 7 --s 28