# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("HaimingW/Leandojo-CodeLLama-7b")
model = AutoModelForCausalLM.from_pretrained("HaimingW/Leandojo-CodeLLama-7b")Quick Links
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
This is a CodeLlama-7b model fine-tuned on LeanDojo Benchmark
The performance is comparable with the paper Temperature-scaled large language models for Lean proofstep prediction with around 57.7 pass@1 accuracy on LeanDojo "random" benchmark.
The training lines are formatted as follow:
GOAL $(tactic state) PROOFSTEP $(tactic)
For inference, use the following line:
GOAL $(tactic state) PROOFSTEP
- Downloads last month
- 10
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="HaimingW/Leandojo-CodeLLama-7b")