BTGenBot
Collection
4 items • Updated • 1
How to use AIRLab-POLIMI/llama-2-7b-chat-hf-btgenbot-adapter with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
model = PeftModel.from_pretrained(base_model, "AIRLab-POLIMI/llama-2-7b-chat-hf-btgenbot-adapter")LoRA adapter release for the paper BTGenBot: Behavior Tree Generation for Robotic Tasks with Lightweight LLMs, currently in submission at IEEE/RSJ International Conference on Intelligent Robots and Systems.GitHub RepositoryPaper
Behavior trees generation for robotic tasks
The following bitsandbytes quantization config was used during training:
Base model
meta-llama/Llama-2-7b-chat-hf