snehal1729 commited on
Commit
8d76d2e
1 Parent(s): a898ef1

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +21 -32
model_index.json CHANGED
@@ -1,33 +1,22 @@
1
- {
2
- "_class_name": "StableDiffusionInstructPix2PixPipeline",
3
- "_diffusers_version": "0.12.0.dev0",
4
- "feature_extractor": [
5
- "transformers",
6
- "CLIPImageProcessor"
7
- ],
8
- "requires_safety_checker": false,
9
- "safety_checker": [
10
- "stable_diffusion",
11
- "StableDiffusionSafetyChecker"
12
- ],
13
- "scheduler": [
14
- "diffusers",
15
- "EulerAncestralDiscreteScheduler"
16
- ],
17
- "text_encoder": [
18
- "transformers",
19
- "CLIPTextModel"
20
- ],
21
- "tokenizer": [
22
- "transformers",
23
- "CLIPTokenizer"
24
- ],
25
- "unet": [
26
- "diffusers",
27
- "UNet2DConditionModel"
28
- ],
29
- "vae": [
30
- "diffusers",
31
- "AutoencoderKL"
32
- ]
33
  }
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "model_type": "pix2pix",
4
+ "sample_size": 256,
5
+ "in_channels": 3,
6
+ "out_channels": 3,
7
+ "layers_per_block": 2,
8
+ "block_out_channels": [128, 256, 512, 512],
9
+ "down_block_types": [
10
+ "DownBlock2D",
11
+ "DownBlock2D",
12
+ "DownBlock2D",
13
+ "AttnDownBlock2D"
14
+ ],
15
+ "up_block_types": [
16
+ "AttnUpBlock2D",
17
+ "UpBlock2D",
18
+ "UpBlock2D",
19
+ "UpBlock2D"
20
+ ],
21
+ "conditional": true
 
 
 
 
 
 
 
 
 
 
 
22
  }