hiyouga's picture
Update README.md
84d515d
metadata
license: other
language:
  - en
  - zh
library_name: transformers
pipeline_tag: text-generation
inference: false
tags:
  - baichuan
  - llama2
  - baichuan2

This is the LLaMAfied version of Baichuan2-7B-Base model by Baichuan Inc.

This model is converted with https://github.com/hiyouga/LLaMA-Efficient-Tuning/blob/main/tests/llamafy_baichuan2.py

You may use this model for fine-tuning in downstream tasks, we recommend using our efficient fine-tuning toolkit. https://github.com/hiyouga/LLaMA-Efficient-Tuning

  • Developed by: Baichuan Inc.
  • Language(s) (NLP): Chinese/English
  • License: Baichuan2 License

Usage:

from transformers import AutoModelForCausalLM, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("hiyouga/Baichuan2-7B-Base-LLaMAfied", use_fast=False)
model = AutoModelForCausalLM.from_pretrained("hiyouga/Baichuan2-7B-Base-LLaMAfied").cuda()