Commit
·
7b64305
0
Parent(s):
init
Browse files- .gitattributes +25 -0
- args.json +60 -0
- feature_extractor/preprocessor_config.json +3 -0
- model.ckpt +3 -0
- model_index.json +33 -0
- samples/0.png +3 -0
- samples/1.png +3 -0
- samples/2.png +3 -0
- samples/3.png +3 -0
- scheduler/scheduler_config.json +3 -0
- text_encoder/config.json +3 -0
- text_encoder/pytorch_model.bin +3 -0
- tokenizer/merges.txt +3 -0
- tokenizer/special_tokens_map.json +3 -0
- tokenizer/tokenizer_config.json +3 -0
- tokenizer/vocab.json +3 -0
- unet/config.json +3 -0
- unet/diffusion_pytorch_model.bin +3 -0
- vae/config.json +3 -0
- vae/diffusion_pytorch_model.bin +3 -0
.gitattributes
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
tokenizer/merges.txt filter=lfs diff=lfs merge=lfs -text
|
2 |
+
tokenizer/special_tokens_map.json filter=lfs diff=lfs merge=lfs -text
|
3 |
+
unet/diffusion_pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
+
vae/diffusion_pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
|
5 |
+
scheduler/scheduler_config.json filter=lfs diff=lfs merge=lfs -text
|
6 |
+
tokenizer/tokenizer_config.json filter=lfs diff=lfs merge=lfs -text
|
7 |
+
tokenizer/vocab.json filter=lfs diff=lfs merge=lfs -text
|
8 |
+
unet/config.json filter=lfs diff=lfs merge=lfs -text
|
9 |
+
feature_extractor/preprocessor_config.json filter=lfs diff=lfs merge=lfs -text
|
10 |
+
samples/3.png filter=lfs diff=lfs merge=lfs -text
|
11 |
+
text_encoder/config.json filter=lfs diff=lfs merge=lfs -text
|
12 |
+
vae/config.json filter=lfs diff=lfs merge=lfs -text
|
13 |
+
samples/0.png filter=lfs diff=lfs merge=lfs -text
|
14 |
+
samples/1.png filter=lfs diff=lfs merge=lfs -text
|
15 |
+
samples/2.png filter=lfs diff=lfs merge=lfs -text
|
16 |
+
model.ckpt filter=lfs diff=lfs merge=lfs -text
|
17 |
+
text_encoder/pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
|
18 |
+
tokenizer/* filter=lfs diff=lfs merge=lfs -text
|
19 |
+
unet/* filter=lfs diff=lfs merge=lfs -text
|
20 |
+
vae/* filter=lfs diff=lfs merge=lfs -text
|
21 |
+
feature_extractor/* filter=lfs diff=lfs merge=lfs -text
|
22 |
+
samples/* filter=lfs diff=lfs merge=lfs -text
|
23 |
+
scheduler/* filter=lfs diff=lfs merge=lfs -text
|
24 |
+
text_encoder/* filter=lfs diff=lfs merge=lfs -text
|
25 |
+
fdf filter=lfs diff=lfs merge=lfs -text
|
args.json
ADDED
@@ -0,0 +1,60 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"pretrained_model_name_or_path": "runwayml/stable-diffusion-v1-5",
|
3 |
+
"pretrained_vae_name_or_path": "stabilityai/sd-vae-ft-mse",
|
4 |
+
"revision": "fp16",
|
5 |
+
"tokenizer_name": null,
|
6 |
+
"instance_data_dir": null,
|
7 |
+
"class_data_dir": null,
|
8 |
+
"instance_prompt": null,
|
9 |
+
"class_prompt": null,
|
10 |
+
"save_sample_prompt": "A [V] cat",
|
11 |
+
"save_sample_negative_prompt": null,
|
12 |
+
"n_save_sample": 4,
|
13 |
+
"save_guidance_scale": 7.5,
|
14 |
+
"save_infer_steps": 20,
|
15 |
+
"pad_tokens": false,
|
16 |
+
"with_prior_preservation": true,
|
17 |
+
"prior_loss_weight": 1.0,
|
18 |
+
"num_class_images": 12,
|
19 |
+
"output_dir": "./",
|
20 |
+
"seed": 42,
|
21 |
+
"resolution": 512,
|
22 |
+
"center_crop": false,
|
23 |
+
"train_text_encoder": true,
|
24 |
+
"train_batch_size": 8,
|
25 |
+
"sample_batch_size": 4,
|
26 |
+
"num_train_epochs": 180,
|
27 |
+
"max_train_steps": 360,
|
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,
|
38 |
+
"adam_epsilon": 1e-08,
|
39 |
+
"max_grad_norm": 1.0,
|
40 |
+
"push_to_hub": false,
|
41 |
+
"hub_token": null,
|
42 |
+
"hub_model_id": null,
|
43 |
+
"logging_dir": "logs",
|
44 |
+
"log_interval": 10,
|
45 |
+
"save_interval": 40,
|
46 |
+
"save_min_steps": 0,
|
47 |
+
"mixed_precision": "fp16",
|
48 |
+
"not_cache_latents": false,
|
49 |
+
"hflip": false,
|
50 |
+
"local_rank": -1,
|
51 |
+
"concepts_list": [
|
52 |
+
{
|
53 |
+
"instance_prompt": "A [V] cat",
|
54 |
+
"class_prompt": "A cat",
|
55 |
+
"instance_data_dir": "data/cat_V",
|
56 |
+
"class_data_dir": "data/cat"
|
57 |
+
}
|
58 |
+
],
|
59 |
+
"read_prompts_from_txts": false
|
60 |
+
}
|
feature_extractor/preprocessor_config.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6f638fb9401a6d6296feff533ee7efe657b787c49f954f82f5906b36ef2a1b1f
|
3 |
+
size 520
|
model.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6c6d10a6f1230f73076bc260125384aae68989843a0cae00fbbb17b3e5e5755e
|
3 |
+
size 2132790871
|
model_index.json
ADDED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "StableDiffusionPipeline",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"feature_extractor": [
|
5 |
+
"transformers",
|
6 |
+
"CLIPFeatureExtractor"
|
7 |
+
],
|
8 |
+
"requires_safety_checker": true,
|
9 |
+
"safety_checker": [
|
10 |
+
null,
|
11 |
+
null
|
12 |
+
],
|
13 |
+
"scheduler": [
|
14 |
+
"diffusers",
|
15 |
+
"PNDMScheduler"
|
16 |
+
],
|
17 |
+
"text_encoder": [
|
18 |
+
"transformers",
|
19 |
+
"CLIPTextModel"
|
20 |
+
],
|
21 |
+
"tokenizer": [
|
22 |
+
"transformers",
|
23 |
+
"CLIPTokenizer"
|
24 |
+
],
|
25 |
+
"unet": [
|
26 |
+
"diffusers",
|
27 |
+
"UNet2DConditionModel"
|
28 |
+
],
|
29 |
+
"vae": [
|
30 |
+
"diffusers",
|
31 |
+
"AutoencoderKL"
|
32 |
+
]
|
33 |
+
}
|
samples/0.png
ADDED
![]() |
Git LFS Details
|
samples/1.png
ADDED
![]() |
Git LFS Details
|
samples/2.png
ADDED
![]() |
Git LFS Details
|
samples/3.png
ADDED
![]() |
Git LFS Details
|
scheduler/scheduler_config.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:35442fb22160d0123e0d2f8ee345c5afe08c2601ab6ca7dff870f8befc228e0d
|
3 |
+
size 465
|
text_encoder/config.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:876d05e539e5be52aa52af90fb51ef2f004682aa70d850047821d5c5712b17aa
|
3 |
+
size 613
|
text_encoder/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bfe014be9d3c30fb4f65b31387251cbbdf205e172770565fb693e68e5ee74912
|
3 |
+
size 492308829
|
tokenizer/merges.txt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9fd691f7c8039210e0fced15865466c65820d09b63988b0174bfe25de299051a
|
3 |
+
size 524619
|
tokenizer/special_tokens_map.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c4864a9376a8401918425bed71fc14fc0e81f9b59ec45c1cf96cccb2df508eac
|
3 |
+
size 472
|
tokenizer/tokenizer_config.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:19d7b034cb0cc3ce9766c2231373ab8aa8991fc72e2c8f76558bfaae3de0d563
|
3 |
+
size 737
|
tokenizer/vocab.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e089ad92ba36837a0d31433e555c8f45fe601ab5c221d4f607ded32d9f7a4349
|
3 |
+
size 1059962
|
unet/config.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9a1c2e0cf0a1ee389a342eef631c888fa6ee6acc8214b2da5424234e5a187dd8
|
3 |
+
size 1304
|
unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:89250e7148a32f27bc735a964c91aeddcd0a44f04f6f8a7afa31a5fabf42a307
|
3 |
+
size 3438375973
|
vae/config.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1c2db19640f9b327932ca743c0c7fa39d427237f9d8f70d5551b243e201f6b58
|
3 |
+
size 630
|
vae/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:36bb8e1b54aba3a0914eb35fba13dcb107e9f18d379d1df2158732cd4bf56a94
|
3 |
+
size 334711857
|