From b885c6f9d2e24b452564add86aea8213282f8003 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Thu, 9 Mar 2023 20:25:21 +0900 Subject: [PATCH] disable annoying warning in CLIP loading --- library/model_util.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/library/model_util.py b/library/model_util.py index 1a715c8f..d1020c05 100644 --- a/library/model_util.py +++ b/library/model_util.py @@ -4,7 +4,7 @@ import math import os import torch -from transformers import CLIPTextModel, CLIPTokenizer, CLIPTextConfig +from transformers import CLIPTextModel, CLIPTokenizer, CLIPTextConfig, logging from diffusers import AutoencoderKL, DDIMScheduler, StableDiffusionPipeline, UNet2DConditionModel from safetensors.torch import load_file, save_file @@ -916,7 +916,11 @@ def load_models_from_stable_diffusion_checkpoint(v2, ckpt_path, dtype=None): info = text_model.load_state_dict(converted_text_encoder_checkpoint) else: converted_text_encoder_checkpoint = convert_ldm_clip_checkpoint_v1(state_dict) + + logging.set_verbosity_error() # don't show annoying warning text_model = CLIPTextModel.from_pretrained("openai/clip-vit-large-patch14") + logging.set_verbosity_warning() + info = text_model.load_state_dict(converted_text_encoder_checkpoint) print("loading text encoder:", info)