nliew commited on
Commit
3d17ae7
·
1 Parent(s): 33c0250

Update ONNX model

Browse files
Files changed (1) hide show
  1. model_index.json +14 -6
model_index.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_class_name": "StableDiffusionPipeline",
3
  "_diffusers_version": "0.27.2",
4
  "_name_or_path": "stabilityai/sd-turbo",
5
  "feature_extractor": [
@@ -20,8 +20,12 @@
20
  "EulerDiscreteScheduler"
21
  ],
22
  "text_encoder": [
23
- "transformers",
24
- "CLIPTextModel"
 
 
 
 
25
  ],
26
  "tokenizer": [
27
  "transformers",
@@ -29,10 +33,14 @@
29
  ],
30
  "unet": [
31
  "diffusers",
32
- "UNet2DConditionModel"
 
 
 
 
33
  ],
34
- "vae": [
35
  "diffusers",
36
- "AutoencoderKL"
37
  ]
38
  }
 
1
  {
2
+ "_class_name": "ORTStableDiffusionPipeline",
3
  "_diffusers_version": "0.27.2",
4
  "_name_or_path": "stabilityai/sd-turbo",
5
  "feature_extractor": [
 
20
  "EulerDiscreteScheduler"
21
  ],
22
  "text_encoder": [
23
+ "diffusers",
24
+ "OnnxRuntimeModel"
25
+ ],
26
+ "text_encoder_2": [
27
+ null,
28
+ null
29
  ],
30
  "tokenizer": [
31
  "transformers",
 
33
  ],
34
  "unet": [
35
  "diffusers",
36
+ "OnnxRuntimeModel"
37
+ ],
38
+ "vae_decoder": [
39
+ "diffusers",
40
+ "OnnxRuntimeModel"
41
  ],
42
+ "vae_encoder": [
43
  "diffusers",
44
+ "OnnxRuntimeModel"
45
  ]
46
  }