From 7deeff35bcc5279d9b3ba985688d1dd29af634f8 Mon Sep 17 00:00:00 2001 From: Pawel Kowalski Date: Thu, 14 Dec 2023 09:57:55 +0100 Subject: [PATCH] Reverting to a single model --- stable_diffusion/stable_diffusion/model_io.py | 13 +------------ 1 file changed, 1 insertion(+), 12 deletions(-) diff --git a/stable_diffusion/stable_diffusion/model_io.py b/stable_diffusion/stable_diffusion/model_io.py index 349e5d76..57879ef9 100644 --- a/stable_diffusion/stable_diffusion/model_io.py +++ b/stable_diffusion/stable_diffusion/model_io.py @@ -30,18 +30,7 @@ _MODELS = { "diffusion_config": "scheduler/scheduler_config.json", "tokenizer_vocab": "tokenizer/vocab.json", "tokenizer_merges": "tokenizer/merges.txt", - }, - "nitrosocke/Ghibli-Diffusion": { - "unet_config": "unet/config.json", - "unet": "unet/diffusion_pytorch_model.safetensors", - "text_encoder_config": "text_encoder/config.json", - "text_encoder": "text_encoder/model.safetensors", - "vae_config": "vae/config.json", - "vae": "vae/diffusion_pytorch_model.safetensors", - "diffusion_config": "scheduler/scheduler_config.json", - "tokenizer_vocab": "tokenizer/vocab.json", - "tokenizer_merges": "tokenizer/merges.txt", - }, + } }