Llama 3 8B Data Analysis

Published by @dataframe_ai · Community adapter

Data wrangling and analysis assistant trained on Jupyter notebooks from Kaggle and academic preprints. Strong on pandas idioms, matplotlib chart composition, and statistical interpretation.

Usage

from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
base = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
model = PeftModel.from_pretrained(base, "modelforgelab/llama3-8b-data-analysis-lora")

inputs = tokenizer("Your prompt here", return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

Compatibility

• transformers>=4.40
• vLLM