Peng Ziqiao

亲测可以在colab跑，内存占用3.5G左右，显存占用5.5G左右 ``` !pip install --upgrade protobuf icetk cpm_kernels from transformers import AutoTokenizer, AutoModel # 指定缓存位置 os.environ["TRANSFORMERS_CACHE"] = "/content/models_cache" # 加载模型 tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b-int4", trust_remote_code=True) model = AutoModel.from_pretrained("THUDM/chatglm-6b-int4",trust_remote_code=True).half().cuda() model = model.eval()...

I edited 'stable-diffusion-webui/extensions/stable-diffusion-webui-wd14-tagger/tagger/interrogator.py' ```python def download(self) -> Tuple[os.PathLike, os.PathLike]: print(f"Loading {self.name} model file from {self.kwargs['repo_id']}") # model_path = Path(hf_hub_download( # **self.kwargs, filename=self.model_path)) # tags_path = Path(hf_hub_download( # **self.kwargs, filename=self.tags_path)) #...