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

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

@ -32,6 +32,7 @@ from omegaconf import OmegaConf
from tqdm import tqdm from tqdm import tqdm
from transformers import ( from transformers import (
CLIPTextModel, CLIPTextModel,
CLIPTextConfig,
CLIPTokenizer, CLIPTokenizer,
AutoFeatureExtractor, AutoFeatureExtractor,
BertTokenizerFast, BertTokenizerFast,
@ -205,12 +206,14 @@ def download_conversion_models():
pipeline = CLIPTextModel.from_pretrained(repo_id, subfolder="text_encoder", **kwargs) pipeline = CLIPTextModel.from_pretrained(repo_id, subfolder="text_encoder", **kwargs)
pipeline.save_pretrained(target_dir / 'stable-diffusion-2-clip' / 'text_encoder', safe_serialization=True) 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" repo_id = "laion/CLIP-ViT-bigG-14-laion2B-39B-b160k"
_, model_name = repo_id.split('/') _, model_name = repo_id.split('/')
tokenizer_2 = CLIPTokenizer.from_pretrained(repo_id, **kwargs) pipeline = CLIPTokenizer.from_pretrained(repo_id, **kwargs)
tokenizer_2.save_pretrained(target_dir / model_name, safe_serialization=True) pipeline.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 = CLIPTextConfig.from_pretrained(repo_id, **kwargs)
pipeline.save_pretrained(target_dir / model_name, safe_serialization=True)
# VAE # VAE
logger.info('Downloading stable diffusion VAE') logger.info('Downloading stable diffusion VAE')

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

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