模型推荐

rafavi

1. Qwen2.5-Coder（华为的代码模型，兼容性好）
python

1. # 安装
   
2. pip install transformers accelerate
   
3. 
   
4. # 使用
   
5. from transformers import AutoTokenizer, AutoModelForCausalLM
   
6. 
   
7. model_name = "Qwen/Qwen2.5-Coder-7B-Instruct"
   
8. tokenizer = AutoTokenizer.from_pretrained(model_name)
   
9. model = AutoModelForCausalLM.from_pretrained(model_name)

复制代码

2. CodeLlama（Meta的代码模型，非常稳定）
python

1. model_name = "codellama/CodeLlama-7b-Instruct-hf"

复制代码

3. DeepSeek-Coder-V1（V1版本更稳定）
python

1. model_name = "deepseek-ai/deepseek-coder-6.7b-instruct"

复制代码