ahmedghani commited on
Commit
31fa3ea
1 Parent(s): 519d556

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -93
README.md CHANGED
@@ -1,97 +1,13 @@
1
  ---
2
  license: creativeml-openrail-m
 
 
 
 
3
  ---
4
 
5
- ### MODEL_CONFIG_DDIM_TRAIN
6
-
7
- ```
8
- MODEL_CONFIG_DDIM = {
9
- "_class_name": "StableDiffusionPipeline",
10
- "_diffusers_version": "0.6.0",
11
- "scheduler": [
12
- "diffusers",
13
- "DDIMScheduler"
14
- ],
15
- "text_encoder": [
16
- "transformers",
17
- "CLIPTextModel"
18
- ],
19
- "tokenizer": [
20
- "transformers",
21
- "CLIPTokenizer"
22
- ],
23
- "unet": [
24
- "diffusers",
25
- "UNet2DConditionModel"
26
- ],
27
- "vae": [
28
- "diffusers",
29
- "AutoencoderKL"
30
- ]
31
- }
32
- ```
33
-
34
- ### MODEL_CONFIG_DDIM_SAVE
35
- ```
36
- MODEL_CONFIG_DDIM_SAVE = {
37
- "_class_name": "StableDiffusionPipeline",
38
- "_diffusers_version": "0.9.0.dev0",
39
- "scheduler": [
40
- "diffusers",
41
- "DDIMScheduler"
42
- ],
43
- "text_encoder": [
44
- "transformers",
45
- "CLIPTextModel"
46
- ],
47
- "tokenizer": [
48
- "transformers",
49
- "CLIPTokenizer"
50
- ],
51
- "unet": [
52
- "diffusers",
53
- "UNet2DConditionModel"
54
- ],
55
- "vae": [
56
- "diffusers",
57
- "AutoencoderKL"
58
- ]
59
- }
60
- ```
61
-
62
-
63
- ### SCHEDULER_CONFIG_DDIM_TRAIN
64
- ```
65
- SCHEDULER_CONFIG_DDIM_TRAIN = {
66
- "_class_name": "DDIMScheduler",
67
- "_diffusers_version": "0.6.0",
68
- "beta_end": 0.012,
69
- "beta_schedule": "scaled_linear",
70
- "beta_start": 0.00085,
71
- "clip_sample": false,
72
- "num_train_timesteps": 1000,
73
- "set_alpha_to_one": false,
74
- "skip_prk_steps": true,
75
- "steps_offset": 1,
76
- "trained_betas": null
77
-
78
- }
79
-
80
- ```
81
- ### SCHEDULER_CONFIG_DDIM_SAVE
82
- ```
83
- SCHEDULER_CONFIG_DDIM_SAVE = {
84
- "_class_name": "DDIMScheduler",
85
- "_diffusers_version": "0.9.0.dev0",
86
- "beta_end": 0.012,
87
- "beta_schedule": "scaled_linear",
88
- "beta_start": 0.00085,
89
- "clip_sample": false,
90
- "num_train_timesteps": 1000,
91
- "prediction_type": "epsilon",
92
- "set_alpha_to_one": false,
93
- "skip_prk_steps": true,
94
- "steps_offset": 1,
95
- "trained_betas": null
96
- }
97
- ```
 
1
  ---
2
  license: creativeml-openrail-m
3
+ language:
4
+ - en
5
+ library_name: diffusers
6
+ pipeline_tag: text-to-image
7
  ---
8
 
9
+ ## Stable Diffusion custom checkpoint for finetuning
10
+ ### Settings:
11
+ SD-version: 1.5
12
+ vae: stabilityai/sd-vae-ft-mse
13
+ scheduler: DDIM