davidfant commited on
Commit
01bb789
1 Parent(s): 1972860

stable-diffusion-inpainting, 4/5 breast size (2000 steps)

Browse files
args.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "pretrained_model_name_or_path": "/home/fant/repos/shiv-diffusers/output/model/9000",
3
  "pretrained_vae_name_or_path": "stabilityai/sd-vae-ft-mse",
4
  "revision": "fp16",
5
  "tokenizer_name": null,
@@ -7,7 +7,7 @@
7
  "class_data_dir": null,
8
  "instance_prompt": null,
9
  "class_prompt": null,
10
- "save_sample_prompt": "close up of woman wearing bikini top, breast size xnjka|close up of woman wearing bikini top, breast size qkdgj|close up of woman wearing bikini top, breast size usmda|close up of woman wearing bikini top, breast size opqng",
11
  "save_sample_negative_prompt": null,
12
  "n_save_sample": 4,
13
  "save_guidance_scale": 7.5,
@@ -15,16 +15,16 @@
15
  "pad_tokens": true,
16
  "with_prior_preservation": true,
17
  "prior_loss_weight": 1.0,
18
- "num_class_images": 200,
19
- "output_dir": "/home/fant/repos/shiv-diffusers/output/model-from-9000",
20
  "seed": 3434554,
21
  "resolution": 512,
22
  "center_crop": false,
23
  "train_text_encoder": true,
24
  "train_batch_size": 1,
25
  "sample_batch_size": 4,
26
- "num_train_epochs": 25,
27
- "max_train_steps": 20000,
28
  "gradient_accumulation_steps": 1,
29
  "gradient_checkpointing": false,
30
  "learning_rate": 1e-06,
@@ -42,7 +42,7 @@
42
  "hub_model_id": null,
43
  "logging_dir": "logs",
44
  "log_interval": 10,
45
- "save_interval": 1000,
46
  "save_min_steps": 0,
47
  "mixed_precision": "fp16",
48
  "not_cache_latents": true,
@@ -51,26 +51,26 @@
51
  {
52
  "class_prompt": "close up of woman wearing bikini top",
53
  "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
54
- "instance_prompt": "close up of woman wearing bikini top, breast size xnjka",
55
- "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/A"
56
  },
57
  {
58
  "class_prompt": "close up of woman wearing bikini top",
59
  "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
60
- "instance_prompt": "close up of woman wearing bikini top, breast size qkdgj",
61
- "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/B"
62
  },
63
  {
64
  "class_prompt": "close up of woman wearing bikini top",
65
  "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
66
- "instance_prompt": "close up of woman wearing bikini top, breast size usmda",
67
- "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/C"
68
  },
69
  {
70
  "class_prompt": "close up of woman wearing bikini top",
71
  "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
72
- "instance_prompt": "close up of woman wearing bikini top, breast size opqng",
73
- "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/D"
74
  }
75
  ]
76
  }
 
1
  {
2
+ "pretrained_model_name_or_path": "runwayml/stable-diffusion-inpainting",
3
  "pretrained_vae_name_or_path": "stabilityai/sd-vae-ft-mse",
4
  "revision": "fp16",
5
  "tokenizer_name": null,
 
7
  "class_data_dir": null,
8
  "instance_prompt": null,
9
  "class_prompt": null,
10
+ "save_sample_prompt": "close up of woman wearing bikini top, amfdk breast size|close up of woman wearing bikini top, boqnf breast size|close up of woman wearing bikini top, czufm breast size|close up of woman wearing bikini top, dpqjd breast size",
11
  "save_sample_negative_prompt": null,
12
  "n_save_sample": 4,
13
  "save_guidance_scale": 7.5,
 
15
  "pad_tokens": true,
16
  "with_prior_preservation": true,
17
  "prior_loss_weight": 1.0,
18
+ "num_class_images": 300,
19
+ "output_dir": "/home/fant/repos/model/breast-size",
20
  "seed": 3434554,
21
  "resolution": 512,
22
  "center_crop": false,
23
  "train_text_encoder": true,
24
  "train_batch_size": 1,
25
  "sample_batch_size": 4,
26
+ "num_train_epochs": 27,
27
+ "max_train_steps": 32000,
28
  "gradient_accumulation_steps": 1,
29
  "gradient_checkpointing": false,
30
  "learning_rate": 1e-06,
 
42
  "hub_model_id": null,
43
  "logging_dir": "logs",
44
  "log_interval": 10,
45
+ "save_interval": 2000,
46
  "save_min_steps": 0,
47
  "mixed_precision": "fp16",
48
  "not_cache_latents": true,
 
51
  {
52
  "class_prompt": "close up of woman wearing bikini top",
53
  "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
54
+ "instance_prompt": "close up of woman wearing bikini top, amfdk breast size",
55
+ "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/0"
56
  },
57
  {
58
  "class_prompt": "close up of woman wearing bikini top",
59
  "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
60
+ "instance_prompt": "close up of woman wearing bikini top, boqnf breast size",
61
+ "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/1"
62
  },
63
  {
64
  "class_prompt": "close up of woman wearing bikini top",
65
  "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
66
+ "instance_prompt": "close up of woman wearing bikini top, czufm breast size",
67
+ "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/2"
68
  },
69
  {
70
  "class_prompt": "close up of woman wearing bikini top",
71
  "class_data_dir": "./data/class/close-up-of-woman-wearing-bikini-top",
72
+ "instance_prompt": "close up of woman wearing bikini top, dpqjd breast size",
73
+ "instance_data_dir": "./data/instance/close-up-of-woman-wearing-bikini-top/3"
74
  }
75
  ]
76
  }
samples/close-up-of-woman-wearing-bikini-top,-amfdk-breast-size/0.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-amfdk-breast-size/1.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-amfdk-breast-size/2.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-amfdk-breast-size/3.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-boqnf-breast-size/0.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-boqnf-breast-size/1.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-boqnf-breast-size/2.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-boqnf-breast-size/3.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-czufm-breast-size/0.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-czufm-breast-size/1.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-czufm-breast-size/2.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-czufm-breast-size/3.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-dpqjd-breast-size/0.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-dpqjd-breast-size/1.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-dpqjd-breast-size/2.png ADDED
samples/close-up-of-woman-wearing-bikini-top,-dpqjd-breast-size/3.png ADDED
text_encoder/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/home/fant/repos/shiv-diffusers/output/model/9000",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
 
1
  {
2
+ "_name_or_path": "runwayml/stable-diffusion-inpainting",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
text_encoder/pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:521a9dfd0b17c614d1b70772fcf914d980bb9346449a5c3fd20616e350cbbd7d
3
  size 492308087
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea4823021d0e5f92985662bb2b3f94f4f0ae43c6b39f791200bf160bd6d4f164
3
  size 492308087
tokenizer/tokenizer_config.json CHANGED
@@ -19,7 +19,7 @@
19
  },
20
  "errors": "replace",
21
  "model_max_length": 77,
22
- "name_or_path": "/home/fant/repos/shiv-diffusers/output/model/9000/tokenizer",
23
  "pad_token": "<|endoftext|>",
24
  "special_tokens_map_file": "./special_tokens_map.json",
25
  "tokenizer_class": "CLIPTokenizer",
 
19
  },
20
  "errors": "replace",
21
  "model_max_length": 77,
22
+ "name_or_path": "/home/fant/.cache/huggingface/diffusers/models--runwayml--stable-diffusion-inpainting/snapshots/1855202d3acfad9fd7e30e39d06b771d4b859a18/tokenizer",
23
  "pad_token": "<|endoftext|>",
24
  "special_tokens_map_file": "./special_tokens_map.json",
25
  "tokenizer_class": "CLIPTokenizer",
unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.7.0.dev0",
4
- "_name_or_path": "/home/fant/repos/shiv-diffusers/output/model/9000",
5
  "act_fn": "silu",
6
  "attention_head_dim": 8,
7
  "block_out_channels": [
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.7.0.dev0",
4
+ "_name_or_path": "runwayml/stable-diffusion-inpainting",
5
  "act_fn": "silu",
6
  "attention_head_dim": 8,
7
  "block_out_channels": [
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1eb440a2897199176c9cabf11abd0479be1608a78d55205e8c4c3542e85b400a
3
  size 3438421925
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:21bd0309568111fb6fe2651748b21a780efc66216ff72fa3bb7069e3fa86bbda
3
  size 3438421925