Create index.json
{
"_class_name": "StableDiffusionPipeline",
"architecture": "UNet2DConditionModel",
"_diffusers_version": "0.18.0",
"model_type": "stable-diffusion",
"scheduler": {
"name": "PNDMScheduler",
"scheduler_type": "pndm",
"beta_start": 0.00085,
"beta_end": 0.012,
"beta_schedule": "scaled_linear",
"skip_prk_steps": true,
"set_alpha_to_one": false,
"steps_offset": 1
},
"feature_extractor": {
"feature_size": 64,
"crop_size": 512,
"do_center_crop": true,
"do_normalize": true,
"do_resize": true
},
"safety_checker": {
"architectures": ["StableDiffusionSafetyChecker"],
"clip_feature_length": 768,
"safety_concept_embeds_path": "safety_concept_embeds.pt",
"vision_hidden_size": 768
},
"text_encoder": {
"architectures": ["CLIPTextModel"],
"max_position_embeddings": 77,
"hidden_size": 768,
"intermediate_size": 3072,
"num_attention_heads": 12,
"num_hidden_layers": 12
},
"unet": {
"in_channels": 4,
"down_block_types": [
"CrossAttnDownBlock2D",
"CrossAttnDownBlock2D",
"CrossAttnDownBlock2D",
"DownBlock2D"
],
"up_block_types": [
"UpBlock2D",
"CrossAttnUpBlock2D",
"CrossAttnUpBlock2D",
"CrossAttnUpBlock2D"
],
"block_out_channels": [320, 640, 1280, 1280],
"layers_per_block": 2,
"cross_attention_dim": 768,
"attention_head_dim": 8
},
"vae": {
"architectures": ["AutoencoderKL"],
"block_out_channels": [128, 256, 512, 512],
"channels": 4,
"down_block_types": [
"DownEncoderBlock2D",
"DownEncoderBlock2D",
"DownEncoderBlock2D",
"DownEncoderBlock2D"
],
"in_channels": 3,
"latent_channels": 4,
"layers_per_block": 2,
"out_channels": 3,
"up_block_types": [
"UpDecoderBlock2D",
"UpDecoderBlock2D",
"UpDecoderBlock2D",
"UpDecoderBlock2D"
]
}
}
Closed, duplicate of #3