找不到gemma_config.py

#2
by dunwu - opened

使用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文件的。

  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