diffusers support (#2)

- diffusers support (96141cc0aa9646caf0f83acf1f0459bcbb5239d1)
- diffusers support (885d93fcbb82e254527d73c27d838039e555ae87)

model_index.json

@@ -1,5 +1,5 @@
 {
-  "_class_name": "PlaygroundV2dot5Pipeline",
+  "_class_name": "StableDiffusionXLPipeline",
   "_diffusers_version": "0.27.0.dev0",
   "feature_extractor": [
     null,
@@ -12,7 +12,7 @@
   ],
   "scheduler": [
     "diffusers",
-    "EulerDiscreteScheduler"
+    "EDMEulerScheduler"
   ],
   "text_encoder": [
     "transformers",

scheduler/scheduler_config.json

@@ -1,22 +1,10 @@
 {
-  "_class_name": "EulerDiscreteScheduler",
+  "_class_name": "EDMEulerScheduler",
   "_diffusers_version": "0.27.0.dev0",
-  "beta_end": 0.012,
-  "beta_schedule": "scaled_linear",
-  "beta_start": 0.00085,
-  "clip_sample": false,
-  "interpolation_type": "linear",
   "num_train_timesteps": 1000,
   "prediction_type": "epsilon",
-  "rescale_betas_zero_snr": false,
-  "sample_max_value": 1.0,
-  "set_alpha_to_one": false,
-  "sigma_max": null,
-  "sigma_min": null,
-  "skip_prk_steps": true,
-  "steps_offset": 1,
-  "timestep_spacing": "leading",
-  "timestep_type": "discrete",
-  "trained_betas": null,
-  "use_karras_sigmas": false
+  "rho": 7.0,
+  "sigma_data": 0.5,
+  "sigma_max": 80.0,
+  "sigma_min": 0.002
 }

vae/config.json

@@ -21,24 +21,23 @@
   "norm_num_groups": 32,
   "out_channels": 3,
   "sample_size": 1024,
-  "scaling_factor": 0.13025,
   "up_block_types": [
     "UpDecoderBlock2D",
     "UpDecoderBlock2D",
     "UpDecoderBlock2D",
     "UpDecoderBlock2D"
   ],
-  "edm_mean": [
+  "latents_mean": [
     -1.6574,
     1.886,
     -1.383,
     2.5155
   ],
-  "edm_std": [
+  "latents_std": [
     8.4927,
     5.9022,
     6.5498,
     5.2299
   ],
-  "edm_scale": 0.5
+  "scaling_factor": 0.5
 }