Downloads · 30 days
17
7% of all-time downloads
dara-llama-2-7b
UKPLab/dara-llama-2-7b
dara-llama-2-7b is a text generation model from UKPLab. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as apache-2.0.
This model is a fine-tuned semantic parsing LLM agent for KGQA. We fine-tune the llama-2-7B on our curated reasoning trajectory https://huggingface.co/datasets/UKPLab/dara.
All-time downloads
244
Public
Parameters
6.7B
13.5 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors13.5 GB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- License
- apache-2.0
- Model type
- llama
- Access
- Public
- Created
- Jul 31, 2024
- Updated
- Aug 1, 2024
- SHA
- 99cba97a
Try a prompt
Datasets
Model card
From the Hugging Face model README
DARA: Decomposition-Alignment-Reasoning Autonomous Language Agent for Question Answering over Knowledge Graphs
Model Information
<!-- Provide a quick summary of what the model is/does. -->This model is a fine-tuned semantic parsing LLM agent for KGQA. We fine-tune the llama-2-7B on our curated reasoning trajectory https://huggingface.co/datasets/UKPLab/dara.
Model Usage
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained(
"UKPLab/dara-llama-2-7b",
torch_dtype=torch.float16,
device_map="auto",
cache_dir = "cache"
)
For more information, please check the repository https://github.com/UKPLab/acl2024-DARA
Hyperparameters
- Learning rate: 2e-5
- Batch size: 4
- Training epochs: 10
- Task
- Text Generation
- Library
- transformers
- Type
- llama
- License
- apache-2.0
- Created
- Jul 31, 2024
- Updated
- Aug 1, 2024