Allen commited on
Commit
a2c4d59
·
1 Parent(s): 331b274

chore: Update model_index.json with new scheduler and safety checker

Browse files
Files changed (1) hide show
  1. model_index.json +5 -6
model_index.json CHANGED
@@ -1,14 +1,13 @@
1
  {
2
- "_class_name": "StableDiffusionXLPipeline",
3
  "_diffusers_version": "0.28.2",
4
- "feature_extractor": [null, null],
5
- "force_zeros_for_empty_prompt": true,
6
  "image_encoder": [null, null],
7
- "scheduler": ["diffusers", "EulerDiscreteScheduler"],
 
 
8
  "text_encoder": ["transformers", "CLIPTextModel"],
9
- "text_encoder_2": ["transformers", "CLIPTextModelWithProjection"],
10
  "tokenizer": ["transformers", "CLIPTokenizer"],
11
- "tokenizer_2": ["transformers", "CLIPTokenizer"],
12
  "unet": ["diffusers", "UNet2DConditionModel"],
13
  "vae": ["diffusers", "AutoencoderKL"]
14
  }
 
1
  {
2
+ "_class_name": "StableDiffusionPipeline",
3
  "_diffusers_version": "0.28.2",
4
+ "feature_extractor": ["transformers", "CLIPFeatureExtractor"],
 
5
  "image_encoder": [null, null],
6
+ "requires_safety_checker": true,
7
+ "safety_checker": ["stable_diffusion", "StableDiffusionSafetyChecker"],
8
+ "scheduler": ["diffusers", "PNDMScheduler"],
9
  "text_encoder": ["transformers", "CLIPTextModel"],
 
10
  "tokenizer": ["transformers", "CLIPTokenizer"],
 
11
  "unet": ["diffusers", "UNet2DConditionModel"],
12
  "vae": ["diffusers", "AutoencoderKL"]
13
  }