averad commited on
Commit
f68816a
1 Parent(s): a20c448

Add Onnx Branch

Browse files

Attached are the Onnx model files

model_index.json CHANGED
@@ -1,21 +1,21 @@
1
  {
2
- "_class_name": "StableDiffusionPipeline",
3
- "_diffusers_version": "0.4.1",
4
  "feature_extractor": [
5
  "transformers",
6
  "CLIPFeatureExtractor"
7
  ],
8
  "safety_checker": [
9
- "stable_diffusion",
10
- "StableDiffusionSafetyChecker"
11
  ],
12
  "scheduler": [
13
  "diffusers",
14
- "LMSDiscreteScheduler"
15
  ],
16
  "text_encoder": [
17
- "transformers",
18
- "CLIPTextModel"
19
  ],
20
  "tokenizer": [
21
  "transformers",
@@ -23,7 +23,15 @@
23
  ],
24
  "unet": [
25
  "diffusers",
26
- "UNet2DConditionModel"
 
 
 
 
 
 
 
 
27
  ],
28
  "vae": [
29
  "diffusers",
1
  {
2
+ "_class_name": "StableDiffusionOnnxPipeline",
3
+ "_diffusers_version": "0.5.0",
4
  "feature_extractor": [
5
  "transformers",
6
  "CLIPFeatureExtractor"
7
  ],
8
  "safety_checker": [
9
+ "diffusers",
10
+ "OnnxRuntimeModel"
11
  ],
12
  "scheduler": [
13
  "diffusers",
14
+ "PNDMScheduler"
15
  ],
16
  "text_encoder": [
17
+ "diffusers",
18
+ "OnnxRuntimeModel"
19
  ],
20
  "tokenizer": [
21
  "transformers",
23
  ],
24
  "unet": [
25
  "diffusers",
26
+ "OnnxRuntimeModel"
27
+ ],
28
+ "vae_decoder": [
29
+ "diffusers",
30
+ "OnnxRuntimeModel"
31
+ ],
32
+ "vae_encoder": [
33
+ "diffusers",
34
+ "OnnxRuntimeModel"
35
  ],
36
  "vae": [
37
  "diffusers",
safety_checker/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05f8d71f06e12ba35212ef104470973b74a79192f723a443f239152c32e23831
3
+ size 1216193649
scheduler/scheduler_config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "LMSDiscreteScheduler",
3
- "_diffusers_version": "0.4.1",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
1
  {
2
  "_class_name": "LMSDiscreteScheduler",
3
+ "_diffusers_version": "0.5.0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
text_encoder/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7bb471d9a2693995e994ade398b8d6e118c6754535233228523665e619c6cdc5
3
+ size 492394398
tokenizer/tokenizer_config.json CHANGED
@@ -19,7 +19,7 @@
19
  },
20
  "errors": "replace",
21
  "model_max_length": 77,
22
- "name_or_path": "openai/clip-vit-large-patch14",
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": "C:\\Users\\avera/.cache\\huggingface\\diffusers\\models--hakurei--waifu-diffusion\\snapshots\\cc452ee4c493761a0c5d6e92394729dbdcd2cb28\\tokenizer",
23
  "pad_token": "<|endoftext|>",
24
  "special_tokens_map_file": "./special_tokens_map.json",
25
  "tokenizer_class": "CLIPTokenizer",
unet/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b617218dc4c29ddf1137ea8776413db7e047573560db098fb1202ebe73eafc01
3
+ size 775566
unet/weights.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96959b69f28cf6fe23ba747d22f390f7ddccf8cc2d690a33069168d520fafa9a
3
+ size 3438083840
vae_decoder/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:500050d2dbd88e78f073d0a5bbd8100117886b1e2a7b160c29d373cd60fd08ce
3
+ size 198021963
vae_encoder/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:55f84641e6266f860391ceb11787585f5778335b9f8b63eb28658e54f4e194d1
3
+ size 136709494