RST Parser with Llama 2
Collection
Collection of the pretrained LoRA weights of Llama2 for RST discourse parsing. • 36 items • Updated • 3
How to use arumaekawa/rstdt-70b-top_down with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-70b-hf")
model = PeftModel.from_pretrained(base_model, "arumaekawa/rstdt-70b-top_down")The following bitsandbytes quantization config was used during training: