redshift-diffusion / model_index.json
camenduru's picture
flax
9e8b504
raw
history blame
572 Bytes
{
"_class_name": "FlaxStableDiffusionPipeline",
"_diffusers_version": "0.8.0.dev0",
"feature_extractor": [
"transformers",
"CLIPFeatureExtractor"
],
"safety_checker": [
"stable_diffusion",
"FlaxStableDiffusionSafetyChecker"
],
"scheduler": [
"diffusers",
"FlaxPNDMScheduler"
],
"text_encoder": [
"transformers",
"FlaxCLIPTextModel"
],
"tokenizer": [
"transformers",
"CLIPTokenizer"
],
"unet": [
"diffusers",
"FlaxUNet2DConditionModel"
],
"vae": [
"diffusers",
"FlaxAutoencoderKL"
]
}