lxlx

用第一个方法加载也不行

4

import os import torch import pandas as pd import numpy as np from torchvision.transforms import ToPILImage from transformers import AutoImageProcessor from flmr import index_custom_collection from flmr import FLMRQueryEncoderTokenizer, FLMRContextEncoderTokenizer, FLMRModelForRetrieval...

我把模型都下载到本地了，为什么加载时还要连接到HF？你知道的国内不能直连HF

2

from transformers import AutoConfig, AutoModel, AutoImageProcessor, AutoTokenizer import torch #%% checkpoint_path = "/data/liuxiang/58/20_项目/0_模型/PreFLMR_ViT-G" image_processor_name = "/data/liuxiang/58/20_项目/0_模型/clip-vit-large-patch14" query_tokenizer = AutoTokenizer.from_pretrained(checkpoint_path, subfolder="query_tokenizer", trust_remote_code=True) context_tokenizer = AutoTokenizer.from_pretrained(checkpoint_path, subfolder="context_tokenizer", trust_remote_code=True) #%% model = AutoModel.from_pretrained(checkpoint_path,...