William2357
commited on
Commit
•
31acb7a
1
Parent(s):
f2ccb26
End of training
Browse files- README.md +4 -3
- logs/dreambooth/1717903691.6629696/events.out.tfevents.1717903691.de80f62ddeaf.542.1 +3 -0
- logs/dreambooth/1717903691.66538/hparams.yml +60 -0
- logs/dreambooth/events.out.tfevents.1717903691.de80f62ddeaf.542.0 +3 -0
- model_index.json +1 -1
- safety_checker/config.json +1 -1
- scheduler/scheduler_config.json +1 -1
- text_encoder/config.json +2 -2
- text_encoder/model.safetensors +2 -2
- tokenizer/tokenizer_config.json +0 -8
- unet/config.json +1 -1
- unet/diffusion_pytorch_model.safetensors +2 -2
- vae/config.json +2 -2
- vae/diffusion_pytorch_model.safetensors +2 -2
README.md
CHANGED
@@ -7,12 +7,13 @@ tags:
|
|
7 |
- diffusers-training
|
8 |
- stable-diffusion
|
9 |
- stable-diffusion-diffusers
|
10 |
-
- stable-diffusion
|
11 |
-
- stable-diffusion-diffusers
|
12 |
-
- text-to-image
|
13 |
- diffusers
|
14 |
- textual_inversion
|
|
|
|
|
15 |
- diffusers-training
|
|
|
|
|
16 |
base_model: runwayml/stable-diffusion-v1-5
|
17 |
inference: true
|
18 |
instance_prompt: a photo of sks dog
|
|
|
7 |
- diffusers-training
|
8 |
- stable-diffusion
|
9 |
- stable-diffusion-diffusers
|
|
|
|
|
|
|
10 |
- diffusers
|
11 |
- textual_inversion
|
12 |
+
- text-to-image
|
13 |
+
- dreambooth
|
14 |
- diffusers-training
|
15 |
+
- stable-diffusion
|
16 |
+
- stable-diffusion-diffusers
|
17 |
base_model: runwayml/stable-diffusion-v1-5
|
18 |
inference: true
|
19 |
instance_prompt: a photo of sks dog
|
logs/dreambooth/1717903691.6629696/events.out.tfevents.1717903691.de80f62ddeaf.542.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:14236de364ef3009837a22967961afec974cfcc901ea0236950bd61db6ae75eb
|
3 |
+
size 2746
|
logs/dreambooth/1717903691.66538/hparams.yml
ADDED
@@ -0,0 +1,60 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
adam_beta1: 0.9
|
2 |
+
adam_beta2: 0.999
|
3 |
+
adam_epsilon: 1.0e-08
|
4 |
+
adam_weight_decay: 0.01
|
5 |
+
allow_tf32: false
|
6 |
+
center_crop: false
|
7 |
+
checkpointing_steps: 100000
|
8 |
+
checkpoints_total_limit: null
|
9 |
+
class_data_dir: null
|
10 |
+
class_labels_conditioning: null
|
11 |
+
class_prompt: null
|
12 |
+
dataloader_num_workers: 0
|
13 |
+
enable_xformers_memory_efficient_attention: false
|
14 |
+
gradient_accumulation_steps: 1
|
15 |
+
gradient_checkpointing: true
|
16 |
+
hub_model_id: null
|
17 |
+
hub_token: null
|
18 |
+
instance_data_dir: faces
|
19 |
+
instance_prompt: a photo of sks person
|
20 |
+
learning_rate: 1.0e-06
|
21 |
+
local_rank: -1
|
22 |
+
logging_dir: logs
|
23 |
+
lr_num_cycles: 1
|
24 |
+
lr_power: 1.0
|
25 |
+
lr_scheduler: constant
|
26 |
+
lr_warmup_steps: 0
|
27 |
+
max_grad_norm: 1.0
|
28 |
+
max_train_steps: 1200
|
29 |
+
mixed_precision: null
|
30 |
+
num_class_images: 100
|
31 |
+
num_train_epochs: 400
|
32 |
+
num_validation_images: 4
|
33 |
+
offset_noise: false
|
34 |
+
output_dir: output
|
35 |
+
pre_compute_text_embeddings: false
|
36 |
+
pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
|
37 |
+
prior_generation_precision: null
|
38 |
+
prior_loss_weight: 1.0
|
39 |
+
push_to_hub: true
|
40 |
+
report_to: tensorboard
|
41 |
+
resolution: 512
|
42 |
+
resume_from_checkpoint: null
|
43 |
+
revision: null
|
44 |
+
sample_batch_size: 4
|
45 |
+
scale_lr: false
|
46 |
+
seed: null
|
47 |
+
set_grads_to_none: false
|
48 |
+
skip_save_text_encoder: false
|
49 |
+
snr_gamma: null
|
50 |
+
text_encoder_use_attention_mask: false
|
51 |
+
tokenizer_max_length: null
|
52 |
+
tokenizer_name: null
|
53 |
+
train_batch_size: 1
|
54 |
+
train_text_encoder: false
|
55 |
+
use_8bit_adam: true
|
56 |
+
validation_prompt: null
|
57 |
+
validation_scheduler: DPMSolverMultistepScheduler
|
58 |
+
validation_steps: 100
|
59 |
+
variant: null
|
60 |
+
with_prior_preservation: false
|
logs/dreambooth/events.out.tfevents.1717903691.de80f62ddeaf.542.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8e0c236a071d9dc1eb9818b8504c8682405b2370ac5cb182444f172e62338c19
|
3 |
+
size 100634
|
model_index.json
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
{
|
2 |
"_class_name": "StableDiffusionPipeline",
|
3 |
-
"_diffusers_version": "0.
|
4 |
"_name_or_path": "runwayml/stable-diffusion-v1-5",
|
5 |
"feature_extractor": [
|
6 |
"transformers",
|
|
|
1 |
{
|
2 |
"_class_name": "StableDiffusionPipeline",
|
3 |
+
"_diffusers_version": "0.29.0.dev0",
|
4 |
"_name_or_path": "runwayml/stable-diffusion-v1-5",
|
5 |
"feature_extractor": [
|
6 |
"transformers",
|
safety_checker/config.json
CHANGED
@@ -15,7 +15,7 @@
|
|
15 |
"num_attention_heads": 12
|
16 |
},
|
17 |
"torch_dtype": "float32",
|
18 |
-
"transformers_version": "4.41.
|
19 |
"vision_config": {
|
20 |
"dropout": 0.0,
|
21 |
"hidden_size": 1024,
|
|
|
15 |
"num_attention_heads": 12
|
16 |
},
|
17 |
"torch_dtype": "float32",
|
18 |
+
"transformers_version": "4.41.2",
|
19 |
"vision_config": {
|
20 |
"dropout": 0.0,
|
21 |
"hidden_size": 1024,
|
scheduler/scheduler_config.json
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
{
|
2 |
"_class_name": "PNDMScheduler",
|
3 |
-
"_diffusers_version": "0.
|
4 |
"beta_end": 0.012,
|
5 |
"beta_schedule": "scaled_linear",
|
6 |
"beta_start": 0.00085,
|
|
|
1 |
{
|
2 |
"_class_name": "PNDMScheduler",
|
3 |
+
"_diffusers_version": "0.29.0.dev0",
|
4 |
"beta_end": 0.012,
|
5 |
"beta_schedule": "scaled_linear",
|
6 |
"beta_start": 0.00085,
|
text_encoder/config.json
CHANGED
@@ -20,6 +20,6 @@
|
|
20 |
"pad_token_id": 1,
|
21 |
"projection_dim": 768,
|
22 |
"torch_dtype": "float32",
|
23 |
-
"transformers_version": "4.41.
|
24 |
-
"vocab_size":
|
25 |
}
|
|
|
20 |
"pad_token_id": 1,
|
21 |
"projection_dim": 768,
|
22 |
"torch_dtype": "float32",
|
23 |
+
"transformers_version": "4.41.2",
|
24 |
+
"vocab_size": 49408
|
25 |
}
|
text_encoder/model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:778d02eb9e707c3fbaae0b67b79ea0d1399b52e624fb634f2f19375ae7c047c3
|
3 |
+
size 492265168
|
tokenizer/tokenizer_config.json
CHANGED
@@ -16,14 +16,6 @@
|
|
16 |
"rstrip": false,
|
17 |
"single_word": false,
|
18 |
"special": true
|
19 |
-
},
|
20 |
-
"49408": {
|
21 |
-
"content": "<cat-toy>",
|
22 |
-
"lstrip": false,
|
23 |
-
"normalized": true,
|
24 |
-
"rstrip": false,
|
25 |
-
"single_word": false,
|
26 |
-
"special": false
|
27 |
}
|
28 |
},
|
29 |
"bos_token": "<|startoftext|>",
|
|
|
16 |
"rstrip": false,
|
17 |
"single_word": false,
|
18 |
"special": true
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
19 |
}
|
20 |
},
|
21 |
"bos_token": "<|startoftext|>",
|
unet/config.json
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
{
|
2 |
"_class_name": "UNet2DConditionModel",
|
3 |
-
"_diffusers_version": "0.
|
4 |
"_name_or_path": "runwayml/stable-diffusion-v1-5",
|
5 |
"act_fn": "silu",
|
6 |
"addition_embed_type": null,
|
|
|
1 |
{
|
2 |
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.29.0.dev0",
|
4 |
"_name_or_path": "runwayml/stable-diffusion-v1-5",
|
5 |
"act_fn": "silu",
|
6 |
"addition_embed_type": null,
|
unet/diffusion_pytorch_model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:799568afe83550418fa442cc498c4351cf9e328a41749902a41a88e992b464ea
|
3 |
+
size 3438167536
|
vae/config.json
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
{
|
2 |
"_class_name": "AutoencoderKL",
|
3 |
-
"_diffusers_version": "0.
|
4 |
-
"_name_or_path": "runwayml
|
5 |
"act_fn": "silu",
|
6 |
"block_out_channels": [
|
7 |
128,
|
|
|
1 |
{
|
2 |
"_class_name": "AutoencoderKL",
|
3 |
+
"_diffusers_version": "0.29.0.dev0",
|
4 |
+
"_name_or_path": "/root/.cache/huggingface/hub/models--runwayml--stable-diffusion-v1-5/snapshots/1d0c4ebf6ff58a5caecab40fa1406526bca4b5b9/vae",
|
5 |
"act_fn": "silu",
|
6 |
"block_out_channels": [
|
7 |
128,
|
vae/diffusion_pytorch_model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b4d2b5932bb4151e54e694fd31ccf51fca908223c9485bd56cd0e1d83ad94c49
|
3 |
+
size 334643268
|