llama400m-climblab-roleplay-5k-random-dora-adapter

Adapter only for DoRA fine-tuned LLaMA 400M model on randomly chose 5k data from roleplay dataset using LMFlow

Adapter Details

This is the DoRA adapter for leonzc/llama400m-climblab-roleplay-5k-random-dora-merged.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
# Load the base model first
base_model = AutoModelForCausalLM.from_pretrained("data4elm/Llama-400M-12L")
# Load the DoRA adapter
model = PeftModel.from_pretrained(base_model, "leonzc/llama400m-climblab-roleplay-5k-random-dora-adapter")
# Load the tokenizer from the base model
tokenizer = AutoTokenizer.from_pretrained("data4elm/Llama-400M-12L")
# Example usage
input_text = "What is the capital of France?"
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(inputs.input_ids, max_length=50)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))