zai-org/AgentInstruct
Viewer • Updated • 1.87k • 1.29k • 234
How to use poisson-fish/dolphin-2.1-mistral-7b-agent-lora with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
model = PeftModel.from_pretrained(base_model, "poisson-fish/dolphin-2.1-mistral-7b-agent-lora")This is a rank 64 LoRA finetune of ehartford/dolphin-2.1-mistral-7b on THUDM/AgentInstruct for 1.1 epochs.
Dolphin-Agent uses ChatML as the prompt format:
<|im_start|>system
You are Dolphin, a helpful AI assistant.<|im_end|>
<|im_start|>user
If Danny owns a bike, then Edward owns a bike. If Edward owns a bike, then Freddy owns a bike. If Danny owns a bike, which of the following statements must be true? Let's think step by step.
I. Edward owns a bike.
II. Freddy owns a bike.
III. Freddy does not own a bike.
Choose one answer:
I only
II only
III only
I and II only
I and III only
<|im_end|>
<|im_start|>assistant
The following bitsandbytes quantization config was used during training:
Base model
mistralai/Mistral-7B-v0.1