Hugging Face's logo

BAAI
/
bge-reranker-v2.5-gemma2-lightweight

Text Classification
sentence-transformers
Safetensors
Transformers
multilingual
cost_wise_gemma
text-generation
custom_code
Model card Files
xet
 Community
13

找不到gemma_config.py

#2
by dunwu - opened
Discussion
dunwu

使用FlagEmbedding加载本地模型，报错OSError: We couldn't connect to 'https://huggingface.co' to load this file, couldn't find it in the cached files and it looks like BAAI/bge-reranker-v2.5-gemma2-lightweight is not the path to a directory containing a file named gemma_config.py. 但是在模型目录下是有gemma_config.py文件的。
cfli
 Beijing Academy of Artificial Intelligence org
•
edited Aug 8, 2024
  1. make sure gemma_config.py and gemma_model.py from BAAI/bge-reranker-v2.5-gemma2-lightweight in your local path.
  2. modify the following part of config.json:
"auto_map": {
    "AutoConfig": "gemma_config.CostWiseGemmaConfig",
    "AutoModel": "gemma_model.CostWiseGemmaModel",
    "AutoModelForCausalLM": "gemma_model.CostWiseGemmaForCausalLM"
  },

Sign up or log in to comment