Update config.json

config.json

@@ -1,15 +1,46 @@
 {
-  "architectures": ["UNet2DConditionModel"],
-  "_class_name": "StableDiffusionPipeline",
+  "_class_name": "StableDiffusionXLPipeline",
+  "architectures": [
+    "StableDiffusionXLPipeline"
+  ],
+  "scheduler": [
+    "DPMSolverMultistepScheduler"
+  ],
+  "text_encoder": [
+    "CLIPTextModel"
+  ],
+  "text_encoder_2": [
+    "CLIPTextModelWithProjection"
+  ],
+  "tokenizer": [
+    "CLIPTokenizer"
+  ],
+  "tokenizer_2": [
+    "CLIPTokenizer"
+  ],
+  "unet": [
+    "UNet2DConditionModel"
+  ],
+  "vae": [
+    "AutoencoderKL"
+  ],
+  "force_zeros_for_empty_prompt": true,
+  "add_watermarker": false,
+  "caption_dropout_probability": 0,
+  "clip_skip": null,
+  "requires_safety_checker": false,
+  "force_zeros_for_empty_prompt": true,
+  "requires_aesthetics_score": false,
+  "do_classifier_free_guidance": true,
+  "num_attention_heads": 16,
+  "num_hidden_layers": 32,
+  "projection_dim": 1280,
+  "text_encoder_2_projection_dim": 1280,
+  "cross_attention_dim": 2048,
+  "sample_size": 128,
+  "use_linear_projection": true,
+  "upcast_attention": false,
+  "model_type": "SDXL",
   "_diffusers_version": "0.21.4",
-  "feature_extractor": [
-    "CLIPFeatureExtractor",
-    "openai/clip-vit-base-patch32"
-  ],
-  "safety_checker": null,
-  "scheduler": ["PNDMScheduler"],
-  "text_encoder": ["CLIPTextModel", "openai/clip-vit-base-patch32"],
-  "tokenizer": ["CLIPTokenizer", "openai/clip-vit-base-patch32"],
-  "unet": ["UNet2DConditionModel"],
-  "vae": ["AutoencoderKL"]
+  "transformer_layers_per_block": 2
 }