Yutian010313's picture
Upload 6 files
428eda2 verified
|
raw
history blame
1.82 kB
metadata
license: llama3
language:
  - en
base_model: meta-llama/Meta-Llama-3-8B-Instruct
tags:
  - large language models
  - speech-language models
  - speech interaction
  - speech-to-speech
library_name: Auto-RAG

Auto-RAG: Autonomous Retrieval-Augmented Generation for Large Language models

Tian Yu, Shaolei Zhang, and Yang Feng*

Model Details

  • Discription:
  • Developed by: ICTNLP Group. Authors: Tian Yu, Shaolei Zhang and Yang Feng.
  • Github Repository: https://github.com/ictnlp/Auto-RAG
  • Finetuned from model: Meta-Llama3-8B-Instruct

Uses

Merge the Meta-Llama3-8B-Instruct weights and Adapter weights.

import os
from transformers import AutoTokenizer, LlamaForCausalLM
import torch

model = LlamaForCausalLM.from_pretrained(PATH_TO_META_LLAMA3_8B_INSTRUCT,
                                  device_map="cpu", 
                                  )
from peft import PeftModel

model = PeftModel.from_pretrained(model, 
                                  PATH_TO_ADAPTER)

from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained(PATH_TO_META_LLAMA3_8B_INSTRUCT)

model = model.merge_and_unload()
model.save_pretrained(SAVE_PATH)
tokenizer.save_pretrained(SAVE_PATH)

Subsequently, you can deploy using frameworks such as vllm.

Citation [optional]

BibTeX:

[More Information Needed]