First model run 10k iterations
Browse files- args.json +8 -8
- samples/0.png +0 -0
- samples/1.png +0 -0
- samples/2.png +0 -0
- samples/3.png +0 -0
- text_encoder/config.json +1 -1
- text_encoder/pytorch_model.bin +3 -0
- tokenizer/tokenizer_config.json +1 -1
- unet/config.json +1 -1
- unet/diffusion_pytorch_model.bin +3 -0
- vae/diffusion_pytorch_model.bin +3 -0
args.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"pretrained_model_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/
|
3 |
"pretrained_vae_name_or_path": "stabilityai/sd-vae-ft-mse",
|
4 |
"revision": null,
|
5 |
"tokenizer_name": null,
|
@@ -15,7 +15,7 @@
|
|
15 |
"pad_tokens": false,
|
16 |
"with_prior_preservation": true,
|
17 |
"prior_loss_weight": 1.0,
|
18 |
-
"num_class_images":
|
19 |
"output_dir": "/content/drive/MyDrive/demslay_diffusion_weights/zwx",
|
20 |
"seed": 1337,
|
21 |
"resolution": 512,
|
@@ -23,11 +23,11 @@
|
|
23 |
"train_text_encoder": true,
|
24 |
"train_batch_size": 1,
|
25 |
"sample_batch_size": 4,
|
26 |
-
"num_train_epochs":
|
27 |
-
"max_train_steps":
|
28 |
"gradient_accumulation_steps": 1,
|
29 |
"gradient_checkpointing": false,
|
30 |
-
"learning_rate":
|
31 |
"scale_lr": false,
|
32 |
"lr_scheduler": "polynomial",
|
33 |
"lr_warmup_steps": 0,
|
@@ -42,7 +42,7 @@
|
|
42 |
"hub_model_id": null,
|
43 |
"logging_dir": "logs",
|
44 |
"log_interval": 10,
|
45 |
-
"save_interval":
|
46 |
"save_min_steps": 0,
|
47 |
"mixed_precision": "fp16",
|
48 |
"not_cache_latents": false,
|
@@ -51,9 +51,9 @@
|
|
51 |
"concepts_list": [
|
52 |
{
|
53 |
"instance_prompt": "demslay style",
|
54 |
-
"class_prompt": "
|
55 |
"instance_data_dir": "/content/data/demslay",
|
56 |
-
"class_data_dir": "/content/data/
|
57 |
}
|
58 |
]
|
59 |
}
|
|
|
1 |
{
|
2 |
+
"pretrained_model_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/2000",
|
3 |
"pretrained_vae_name_or_path": "stabilityai/sd-vae-ft-mse",
|
4 |
"revision": null,
|
5 |
"tokenizer_name": null,
|
|
|
15 |
"pad_tokens": false,
|
16 |
"with_prior_preservation": true,
|
17 |
"prior_loss_weight": 1.0,
|
18 |
+
"num_class_images": 100,
|
19 |
"output_dir": "/content/drive/MyDrive/demslay_diffusion_weights/zwx",
|
20 |
"seed": 1337,
|
21 |
"resolution": 512,
|
|
|
23 |
"train_text_encoder": true,
|
24 |
"train_batch_size": 1,
|
25 |
"sample_batch_size": 4,
|
26 |
+
"num_train_epochs": 10,
|
27 |
+
"max_train_steps": 1000,
|
28 |
"gradient_accumulation_steps": 1,
|
29 |
"gradient_checkpointing": false,
|
30 |
+
"learning_rate": 1e-06,
|
31 |
"scale_lr": false,
|
32 |
"lr_scheduler": "polynomial",
|
33 |
"lr_warmup_steps": 0,
|
|
|
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": false,
|
|
|
51 |
"concepts_list": [
|
52 |
{
|
53 |
"instance_prompt": "demslay style",
|
54 |
+
"class_prompt": "person",
|
55 |
"instance_data_dir": "/content/data/demslay",
|
56 |
+
"class_data_dir": "/content/data/person"
|
57 |
}
|
58 |
]
|
59 |
}
|
samples/0.png
CHANGED
![]() |
![]() |
samples/1.png
CHANGED
![]() |
![]() |
samples/2.png
CHANGED
![]() |
![]() |
samples/3.png
CHANGED
![]() |
![]() |
text_encoder/config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/
|
3 |
"architectures": [
|
4 |
"CLIPTextModel"
|
5 |
],
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/2000",
|
3 |
"architectures": [
|
4 |
"CLIPTextModel"
|
5 |
],
|
text_encoder/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7cc7179aa45e6ed69e7c8c802e3adf27cbf43c7ee4e8da00864991b764cd66c9
|
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": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/
|
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": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/2000/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.9.0",
|
4 |
-
"_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/
|
5 |
"act_fn": "silu",
|
6 |
"attention_head_dim": 8,
|
7 |
"block_out_channels": [
|
|
|
1 |
{
|
2 |
"_class_name": "UNet2DConditionModel",
|
3 |
"_diffusers_version": "0.9.0",
|
4 |
+
"_name_or_path": "/content/drive/MyDrive/demslay_diffusion_weights/zwx/2000",
|
5 |
"act_fn": "silu",
|
6 |
"attention_head_dim": 8,
|
7 |
"block_out_channels": [
|
unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d66ee9acbe30a0ca3e502ddc405e2828ce70dafce791f06f409b1b3e14be545f
|
3 |
+
size 3438364325
|
vae/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1b4889b6b1d4ce7ae320a02dedaeff1780ad77d415ea0d744b476155c6377ddc
|
3 |
+
size 334707217
|