Hugging Face's logo

nitrosocke
/
Arcane-Diffusion

Text-to-Image
Diffusers
StableDiffusionPipeline
stable-diffusion
Inference Endpoints
Model card Files Community
32

Add Onnx Branch

#20
by averad - opened
base: refs/heads/main
←
from: refs/pr/20
Discussion Files changed
+36
-13
Files changed (9) hide show
  1. model_index.json +14 -10
  2. safety_checker/model.onnx +3 -0
  3. scheduler/scheduler_config.json +3 -2
  4. text_encoder/model.onnx +3 -0
  5. tokenizer/tokenizer_config.json +1 -1
  6. unet/model.onnx +3 -0
  7. unet/weights.pb +3 -0
  8. vae_decoder/model.onnx +3 -0
  9. vae_encoder/model.onnx +3 -0
model_index.json CHANGED
@@ -1,21 +1,21 @@
1
  {
2
- "_class_name": "StableDiffusionPipeline",
3
- "_diffusers_version": "0.7.0.dev0",
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,10 +23,14 @@
23
  ],
24
  "unet": [
25
  "diffusers",
26
- "UNet2DConditionModel"
 
 
 
 
27
  ],
28
- "vae": [
29
  "diffusers",
30
- "AutoencoderKL"
31
  ]
32
  }
1
  {
2
+ "_class_name": "StableDiffusionOnnxPipeline",
3
+ "_diffusers_version": "0.3.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
  }
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,9 +1,10 @@
1
  {
2
- "_class_name": "LMSDiscreteScheduler",
3
- "_diffusers_version": "0.7.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "num_train_timesteps": 1000,
 
8
  "trained_betas": null
9
  }
1
  {
2
+ "_class_name": "PNDMScheduler",
3
+ "_diffusers_version": "0.3.0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "num_train_timesteps": 1000,
8
+ "skip_prk_steps": true,
9
  "trained_betas": null
10
  }
text_encoder/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb03f624282b1ad89ecb60b74f8783a86cbe024f6d7bf8d14768572f47fcf625
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": "./arcane_diffusion_v3_diffusers\\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:e68dde2647c01e8ddcf708840f75fbfb7b35297bba48db0d9e8cc1459c120978
3
+ size 1022686
unet/weights.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69e746a00a0774660c0404b0bdecf493af3292bc76be81d40ea2565aa23a5258
3
+ size 3438083840
vae_decoder/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:feb727cf864dda6e3b16d51b29829076ddc3aa0c91adb6f6a928b297a61781d4
3
+ size 198025411
vae_encoder/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b847cade95b95655341edef6680ec3ce0f90525d8446c39e1d245d90c8923077
3
+ size 136711934