JiunYi's picture
Update README.md
c50b520
metadata
license: apache-2.0
datasets:
  - shibing624/alpaca-zh
language:
  - zh
tags:
  - LoRA
  - LLaMA
  - Alpaca
  - PEFT
  - int8

Model Card for llama-7b-alpaca-zh-20k

Uses

Direct Use

from peft import PeftModel
from transformers import GenerationConfig, LlamaForCausalLM, LlamaTokenizer


tokenizer = LlamaTokenizer.from_pretrained("decapoda-research/llama-7b-hf")
model = LlamaForCausalLM.from_pretrained(
    "decapoda-research/llama-7b-hf",
    load_in_8bit=True,
    torch_dtype=torch.float16,
    device_map="auto"
)
model = PeftModel.from_pretrained(
    model,
    "DataAgent/llama-7b-alpaca-zh-20k",
    torch_dtype=torch.float16
)