davidfant commited on
Commit
e5c1976
·
1 Parent(s): 7a4ee0d

stable-diffusion-inpainting, one concept with specific prompt, colorful pattern (2400 steps)

Browse files
args.json CHANGED
@@ -7,7 +7,7 @@
7
  "class_data_dir": null,
8
  "instance_prompt": null,
9
  "class_prompt": null,
10
- "save_sample_prompt": "a woman wearing a blue qwe bikini with cyan floral patterns, short blonde hair, on a boat with an island in the background",
11
  "save_sample_negative_prompt": null,
12
  "n_save_sample": 4,
13
  "save_guidance_scale": 7.5,
@@ -15,7 +15,7 @@
15
  "pad_tokens": true,
16
  "with_prior_preservation": true,
17
  "prior_loss_weight": 1.0,
18
- "num_class_images": 50,
19
  "output_dir": "/home/fant/repos/shiv-diffusers/output/model",
20
  "seed": 3434554,
21
  "resolution": 512,
@@ -23,15 +23,15 @@
23
  "train_text_encoder": true,
24
  "train_batch_size": 1,
25
  "sample_batch_size": 4,
26
- "num_train_epochs": 200000,
27
- "max_train_steps": 9999999,
28
  "gradient_accumulation_steps": 1,
29
  "gradient_checkpointing": false,
30
  "learning_rate": 1e-06,
31
  "scale_lr": false,
32
  "lr_scheduler": "constant",
33
  "lr_warmup_steps": 0,
34
- "use_8bit_adam": true,
35
  "adam_beta1": 0.9,
36
  "adam_beta2": 0.999,
37
  "adam_weight_decay": 0.01,
@@ -49,26 +49,21 @@
49
  "local_rank": -1,
50
  "concepts_list": [
51
  {
52
- "url": "https://www.brightswimwear.com/products/bali-top-ocean-dream",
 
 
 
 
 
53
  "image_urls": [
54
- "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1658.jpg?v=1666613045",
55
- "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1664.jpg?v=1666613045",
56
- "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1671.jpg?v=1666613045&width=1946",
57
- "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1680.jpg?v=1666613045&width=1946",
58
- "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-296.jpg?v=1666613045&width=1946",
59
- "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-297.jpg?v=1666613045&width=1946",
60
- "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-299.jpg?v=1666613045&width=1946",
61
- "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-289.jpg?v=1666613045&width=1946",
62
- "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-298.jpg?v=1666613042&width=1946"
63
  ],
64
- "class_name": "bikini",
65
- "class_prompt": "a woman wearing a bikini",
66
- "class_data_dir": "./class/woman/bikini",
67
- "instance_name": "qwe",
68
- "instance_prompt": "a white woman wearing a blue qwe bikini with cyan floral patterns, long dark brown hair",
69
- "instance_data_dir": "./instance/qwe",
70
- "class_mask_data_dir": "./class/woman/bikini-mask",
71
- "instance_mask_data_dir": "./instance/qwe-mask"
72
  }
73
  ]
74
  }
 
7
  "class_data_dir": null,
8
  "instance_prompt": null,
9
  "class_prompt": null,
10
+ "save_sample_prompt": "woman wearing a sks bikini with tzx pattern",
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": 100,
19
  "output_dir": "/home/fant/repos/shiv-diffusers/output/model",
20
  "seed": 3434554,
21
  "resolution": 512,
 
23
  "train_text_encoder": true,
24
  "train_batch_size": 1,
25
  "sample_batch_size": 4,
26
+ "num_train_epochs": 24,
27
+ "max_train_steps": 2400,
28
  "gradient_accumulation_steps": 1,
29
  "gradient_checkpointing": false,
30
  "learning_rate": 1e-06,
31
  "scale_lr": false,
32
  "lr_scheduler": "constant",
33
  "lr_warmup_steps": 0,
34
+ "use_8bit_adam": false,
35
  "adam_beta1": 0.9,
36
  "adam_beta2": 0.999,
37
  "adam_weight_decay": 0.01,
 
49
  "local_rank": -1,
50
  "concepts_list": [
51
  {
52
+ "url": "https://www.brightswimwear.com/collections/all-swim/products/maria-top-sunset",
53
+ "class_prompt": "woman wearing a bikini",
54
+ "class_data_dir": "./data/class/woman-wearing-a-bikini",
55
+ "instance_type": "maria",
56
+ "instance_pattern": "sunset",
57
+ "instance_data_dir": "./data/instance/woman-wearing-a-maria-bikini-sunset-with-sunset-pattern",
58
  "image_urls": [
59
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1345.jpg?v=1666613430",
60
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1337.jpg?v=1666613430",
61
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-206.jpg?v=1666613430",
62
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-219.jpg?v=1666613430",
63
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/StudioSession-222.jpg?v=1666613430",
64
+ "https://cdn.shopify.com/s/files/1/2148/0245/products/DSC1341.jpg?v=1666613430"
 
 
 
65
  ],
66
+ "instance_prompt": "woman wearing a sks bikini with tzx pattern"
 
 
 
 
 
 
 
67
  }
68
  ]
69
  }
samples/0.png CHANGED
samples/1.png CHANGED
samples/2.png CHANGED
samples/3.png CHANGED
text_encoder/pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:02c7a24bf5ca946689727de3a6627ef50af9ac0ea63b976f746ec0cf905600c3
3
  size 492308087
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e1e6cb4d5ed1c495ab9d79b43122bb4104a030780ea7b203e8b52b465f46557
3
  size 492308087
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:90c42317349565625291ae740359000c8d62144b6b665413e832151496ec642e
3
  size 3438421925
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c2c35c8a7b1f74ce1b63f0d0a5270b61ab817b4673bfd72dc193e90dff34a0b4
3
  size 3438421925