model_index.json

#1
by nibo - opened

Can you please add a model_index.json so we can use it via Colab?

I would love to if I knew how. If you know how please tell me, so I can add it quickly!

Here is the example of the Waifu Diffusion model_Index code:

{
"_class_name": "StableDiffusionPipeline",
"_diffusers_version": "0.4.1",
"feature_extractor": [
"transformers",
"CLIPFeatureExtractor"
],
"safety_checker": [
"stable_diffusion",
"StableDiffusionSafetyChecker"
],
"scheduler": [
"diffusers",
"LMSDiscreteScheduler"
],
"text_encoder": [
"transformers",
"CLIPTextModel"
],
"tokenizer": [
"transformers",
"CLIPTokenizer"
],
"unet": [
"diffusers",
"UNet2DConditionModel"
],
"vae": [
"diffusers",
"AutoencoderKL"
]
}

And here is an example of the Arcane model:

{
"_class_name": "StableDiffusionPipeline",
"_diffusers_version": "0.7.0.dev0",
"feature_extractor": [
"transformers",
"CLIPFeatureExtractor"
],
"safety_checker": [
"stable_diffusion",
"StableDiffusionSafetyChecker"
],
"scheduler": [
"diffusers",
"LMSDiscreteScheduler"
],
"text_encoder": [
"transformers",
"CLIPTextModel"
],
"tokenizer": [
"transformers",
"CLIPTokenizer"
],
"unet": [
"diffusers",
"UNet2DConditionModel"
],
"vae": [
"diffusers",
"AutoencoderKL"
]
}

I think in notepad you should only add that code with the Diffusers Version that you used and saves as model_index.json

i'll see what I can do

Sign up or log in to comment