# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("CorticalStack/mistral-7b-tak-stack-dpo")
model = AutoModelForCausalLM.from_pretrained("CorticalStack/mistral-7b-tak-stack-dpo")Quick Links
mistral-7b-tak-stack-dpo
mistral-7b-tak-stack-dpo is a DPO fine-tuned version of mistralai/Mistral-7B-v0.1 using the CorticalStack/tak-stack-dpo dataset.
LoRA
- r: 32
- LoRA alpha: 32
- LoRA dropout: 0.05
Training arguments
- Batch size: 4
- Gradient accumulation steps: 4
- Optimizer: paged_adamw_32bit
- Max steps: 100
- Learning rate: 5e-05
- Learning rate scheduler type: cosine
- Beta: 0.1
- Max prompt length: 1024
- Max length: 1536
- Downloads last month
- 92
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="CorticalStack/mistral-7b-tak-stack-dpo")