Upload 3 files
Browse files- args.json +59 -0
- model.ckpt +3 -0
- model_index.json +25 -0
args.json
ADDED
@@ -0,0 +1,59 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"pretrained_model_name_or_path": "stabilityai/stable-diffusion-2",
|
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": "snqinyi in a field of flowers, delicate linework, perfect composition, award winning",
|
11 |
+
"save_sample_negative_prompt": null,
|
12 |
+
"n_save_sample": 4,
|
13 |
+
"save_guidance_scale": 7.5,
|
14 |
+
"save_infer_steps": 50,
|
15 |
+
"pad_tokens": false,
|
16 |
+
"with_prior_preservation": true,
|
17 |
+
"prior_loss_weight": 1.0,
|
18 |
+
"num_class_images": 50,
|
19 |
+
"output_dir": "/content/drive/MyDrive/stable_diffusion_weights/snqinyi-diffusion-v4",
|
20 |
+
"seed": 1337,
|
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": 100,
|
27 |
+
"max_train_steps": 5000,
|
28 |
+
"gradient_accumulation_steps": 1,
|
29 |
+
"gradient_checkpointing": true,
|
30 |
+
"learning_rate": 2e-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": 500,
|
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": "snqinyi person",
|
54 |
+
"class_prompt": "beautiful anime man in a field, soft pastel colors, delicate linework, cel shading, perfect composition, award winning",
|
55 |
+
"instance_data_dir": "/content/drive/MyDrive/qinyi",
|
56 |
+
"class_data_dir": "/content/data/anime_man"
|
57 |
+
}
|
58 |
+
]
|
59 |
+
}
|
model.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d5fe08ad8ea6a46d3e1864d926f413b5e632508e5ad6167cd56924becfd025a1
|
3 |
+
size 2580250968
|
model_index.json
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "StableDiffusionPipeline",
|
3 |
+
"_diffusers_version": "0.9.0",
|
4 |
+
"requires_safety_checker": false,
|
5 |
+
"scheduler": [
|
6 |
+
"diffusers",
|
7 |
+
"DDIMScheduler"
|
8 |
+
],
|
9 |
+
"text_encoder": [
|
10 |
+
"transformers",
|
11 |
+
"CLIPTextModel"
|
12 |
+
],
|
13 |
+
"tokenizer": [
|
14 |
+
"transformers",
|
15 |
+
"CLIPTokenizer"
|
16 |
+
],
|
17 |
+
"unet": [
|
18 |
+
"diffusers",
|
19 |
+
"UNet2DConditionModel"
|
20 |
+
],
|
21 |
+
"vae": [
|
22 |
+
"diffusers",
|
23 |
+
"AutoencoderKL"
|
24 |
+
]
|
25 |
+
}
|