Chinese-Vicuna/Chinese-Vicuna-lora-7b-belle-and-guanaco-4bit

Model

1

stars

3

commits

1

linked in READMEs

Jun 12, 2023

updated

alpaca
Chinese-Vicuna
llama

README

This is a Chinese instruction-tuning lora checkpoint based on llama-7B from this repo's work. Specially, this is the 4bit version trained with qlora

You can use it like this:

from transformers import LlamaForCausalLM
from peft import PeftModel

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,
    LORA_PATH, # specific checkpoint path from "Chinese-Vicuna/Chinese-Vicuna-lora-7b-belle-and-guanaco"
    torch_dtype=torch.float16,
    device_map={'': 0}
)

Contributors

Facico

3 commits

Chinese-Vicuna/Chinese-Vicuna-lora-7b-belle-and-guanaco-4bit

Model

1

stars

3

commits

1

linked in READMEs

Jun 12, 2023

updated

alpaca
Chinese-Vicuna
llama

README

This is a Chinese instruction-tuning lora checkpoint based on llama-7B from this repo's work. Specially, this is the 4bit version trained with qlora

You can use it like this:

from transformers import LlamaForCausalLM
from peft import PeftModel

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,
    LORA_PATH, # specific checkpoint path from "Chinese-Vicuna/Chinese-Vicuna-lora-7b-belle-and-guanaco"
    torch_dtype=torch.float16,
    device_map={'': 0}
)

Contributors

Facico

3 commits