Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

save text_encoder_2 config, not whole model

Browse Source
This commit is contained in:
Lincoln Stein 2023-07-22 21:33:40 -04:00
parent 5607794dbb
commit b1d7c9b306
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
invokeai/backend/install/invokeai_configure.py
View File

@ -32,6 +32,7 @@ from omegaconf import OmegaConf
from tqdm import tqdm
from transformers import (
CLIPTextModel,
CLIPTextConfig,
CLIPTokenizer,
AutoFeatureExtractor,
BertTokenizerFast,
@ -205,13 +206,15 @@ def download_conversion_models():
pipeline = CLIPTextModel.from_pretrained(repo_id, subfolder="text_encoder", **kwargs)
pipeline.save_pretrained(target_dir / 'stable-diffusion-2-clip' / 'text_encoder', safe_serialization=True)
# sd-xl - tokenizer_2
repo_id = "laion/CLIP-ViT-bigG-14-laion2B-39B-b160k"
_, model_name = repo_id.split('/')
tokenizer_2 = CLIPTokenizer.from_pretrained(repo_id, **kwargs)
tokenizer_2.save_pretrained(target_dir / model_name, safe_serialization=True)
# for some reason config.json never downloads
hf_download_with_resume(repo_id, target_dir / model_name, "config.json")
pipeline = CLIPTokenizer.from_pretrained(repo_id, **kwargs)
pipeline.save_pretrained(target_dir / model_name, safe_serialization=True)
pipeline = CLIPTextConfig.from_pretrained(repo_id, **kwargs)
pipeline.save_pretrained(target_dir / model_name, safe_serialization=True)
# VAE
logger.info('Downloading stable diffusion VAE')
vae = AutoencoderKL.from_pretrained('stabilityai/sd-vae-ft-mse', **kwargs)

2
invokeai/backend/model_management/convert_ckpt_to_diffusers.py
View File

@ -1711,7 +1711,7 @@ def convert_ckpt_to_diffusers(
pipe.save_pretrained(
dump_path,
safe_serialization=is_safetensors_available(),
safe_serialization=is_safetensors_available() and not no_safetensors,
)
def convert_controlnet_to_diffusers(