AdamOswald1 commited on
Commit
d8dc42d
1 Parent(s): 65ae617

Create model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +65 -0
model_index.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "StableDiffusionPipeline",
3
+ "_diffusers_version": "0.2.4",
4
+ "feature_extractor": [
5
+ "transformers",
6
+ "CLIPFeatureExtractor"
7
+ ],
8
+ "safety_checker": [
9
+ "stable_diffusion",
10
+ "StableDiffusionSafetyChecker"
11
+ ],
12
+ "scheduler": [
13
+ "diffusers",
14
+ "PNDMScheduler"
15
+ ],
16
+ "text_encoder": [
17
+ "transformers",
18
+ "CLIPTextModel"
19
+ ],
20
+ "tokenizer": [
21
+ "transformers",
22
+ "CLIPTokenizer"
23
+ ],
24
+ "unet": [
25
+ "diffusers",
26
+ "UNet2DConditionModel"
27
+ ],
28
+ "vae": [
29
+ "diffusers",
30
+ "AutoencoderKL"
31
+ ]
32
+ }
33
+
34
+ {
35
+ "_class_name": "FlaxStableDiffusionPipeline",
36
+ "_diffusers_version": "0.6.0",
37
+ "feature_extractor": [
38
+ "transformers",
39
+ "CLIPFeatureExtractor"
40
+ ],
41
+ "safety_checker": [
42
+ "stable_diffusion",
43
+ "FlaxStableDiffusionSafetyChecker"
44
+ ],
45
+ "scheduler": [
46
+ "diffusers",
47
+ "FlaxPNDMScheduler"
48
+ ],
49
+ "text_encoder": [
50
+ "transformers",
51
+ "FlaxCLIPTextModel"
52
+ ],
53
+ "tokenizer": [
54
+ "transformers",
55
+ "CLIPTokenizer"
56
+ ],
57
+ "unet": [
58
+ "diffusers",
59
+ "FlaxUNet2DConditionModel"
60
+ ],
61
+ "vae": [
62
+ "diffusers",
63
+ "FlaxAutoencoderKL"
64
+ ]
65
+ }