Instructions to use HanningZhang/Llama3.1-RAG-Reward with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use HanningZhang/Llama3.1-RAG-Reward with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="HanningZhang/Llama3.1-RAG-Reward")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("HanningZhang/Llama3.1-RAG-Reward") model = AutoModelForSequenceClassification.from_pretrained("HanningZhang/Llama3.1-RAG-Reward", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload LlamaForSequenceClassification
Browse files- config.json +1 -1
- model-00004-of-00004.safetensors +2 -2
- model.safetensors.index.json +3 -3
config.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
{
|
| 2 |
"_name_or_path": "llama3_o3_rm/last_checkpoint",
|
| 3 |
"architectures": [
|
| 4 |
-
"
|
| 5 |
],
|
| 6 |
"attention_bias": false,
|
| 7 |
"attention_dropout": 0.0,
|
|
|
|
| 1 |
{
|
| 2 |
"_name_or_path": "llama3_o3_rm/last_checkpoint",
|
| 3 |
"architectures": [
|
| 4 |
+
"LlamaForSequenceClassification"
|
| 5 |
],
|
| 6 |
"attention_bias": false,
|
| 7 |
"attention_dropout": 0.0,
|
model-00004-of-00004.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:06b9df129225c30f3cfc965871a855561b6f16c54c2636680ea68df3a0f05a7c
|
| 3 |
+
size 117473824
|
model.safetensors.index.json
CHANGED
|
@@ -1,9 +1,8 @@
|
|
| 1 |
{
|
| 2 |
"metadata": {
|
| 3 |
-
"total_size":
|
| 4 |
},
|
| 5 |
"weight_map": {
|
| 6 |
-
"lm_head.weight": "model-00004-of-00004.safetensors",
|
| 7 |
"model.embed_tokens.weight": "model-00001-of-00004.safetensors",
|
| 8 |
"model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 9 |
"model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
|
@@ -293,6 +292,7 @@
|
|
| 293 |
"model.layers.9.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 294 |
"model.layers.9.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
| 295 |
"model.layers.9.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
| 296 |
-
"model.norm.weight": "model-00004-of-00004.safetensors"
|
|
|
|
| 297 |
}
|
| 298 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"metadata": {
|
| 3 |
+
"total_size": 15009865728
|
| 4 |
},
|
| 5 |
"weight_map": {
|
|
|
|
| 6 |
"model.embed_tokens.weight": "model-00001-of-00004.safetensors",
|
| 7 |
"model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 8 |
"model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
|
|
|
| 292 |
"model.layers.9.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 293 |
"model.layers.9.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
| 294 |
"model.layers.9.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
| 295 |
+
"model.norm.weight": "model-00004-of-00004.safetensors",
|
| 296 |
+
"score.weight": "model-00004-of-00004.safetensors"
|
| 297 |
}
|
| 298 |
}
|