From db0996c299fdd559ebf9cd48f9dbe47474ed7b07 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Thu, 13 Oct 2022 09:45:27 +0200 Subject: Added TI+Dreambooth training --- infer.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'infer.py') diff --git a/infer.py b/infer.py index 5bd4abc..63b16d8 100644 --- a/infer.py +++ b/infer.py @@ -205,10 +205,10 @@ def load_embeddings(tokenizer, text_encoder, embeddings_dir): def create_pipeline(model, scheduler, embeddings_dir, dtype): print("Loading Stable Diffusion pipeline...") - tokenizer = CLIPTokenizer.from_pretrained(model, subfolder='/tokenizer', torch_dtype=dtype) - text_encoder = CLIPTextModel.from_pretrained(model, subfolder='/text_encoder', torch_dtype=dtype) - vae = AutoencoderKL.from_pretrained(model, subfolder='/vae', torch_dtype=dtype) - unet = UNet2DConditionModel.from_pretrained(model, subfolder='/unet', torch_dtype=dtype) + tokenizer = CLIPTokenizer.from_pretrained(model, subfolder='tokenizer', torch_dtype=dtype) + text_encoder = CLIPTextModel.from_pretrained(model, subfolder='text_encoder', torch_dtype=dtype) + vae = AutoencoderKL.from_pretrained(model, subfolder='vae', torch_dtype=dtype) + unet = UNet2DConditionModel.from_pretrained(model, subfolder='unet', torch_dtype=dtype) load_embeddings(tokenizer, text_encoder, embeddings_dir) -- cgit v1.2.3-54-g00ecf