Text Classification
Transformers
Safetensors
llama
Generated from Trainer
trl
reward-trainer
text-embeddings-inference
Instructions to use bikingSolo/trainer_output with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use bikingSolo/trainer_output with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="bikingSolo/trainer_output", device_map="auto")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("bikingSolo/trainer_output") model = AutoModelForSequenceClassification.from_pretrained("bikingSolo/trainer_output", device_map="auto") - Notebooks
- Google Colab
- Kaggle
bikingSolo/vk-nlp-course-hometask-2-reward-model
Browse files- config.json +6 -0
- model.safetensors +2 -2
- training_args.bin +1 -1
config.json
CHANGED
|
@@ -9,8 +9,14 @@
|
|
| 9 |
"head_dim": 64,
|
| 10 |
"hidden_act": "silu",
|
| 11 |
"hidden_size": 576,
|
|
|
|
|
|
|
|
|
|
| 12 |
"initializer_range": 0.02,
|
| 13 |
"intermediate_size": 1536,
|
|
|
|
|
|
|
|
|
|
| 14 |
"max_position_embeddings": 2048,
|
| 15 |
"mlp_bias": false,
|
| 16 |
"model_type": "llama",
|
|
|
|
| 9 |
"head_dim": 64,
|
| 10 |
"hidden_act": "silu",
|
| 11 |
"hidden_size": 576,
|
| 12 |
+
"id2label": {
|
| 13 |
+
"0": "LABEL_0"
|
| 14 |
+
},
|
| 15 |
"initializer_range": 0.02,
|
| 16 |
"intermediate_size": 1536,
|
| 17 |
+
"label2id": {
|
| 18 |
+
"LABEL_0": 0
|
| 19 |
+
},
|
| 20 |
"max_position_embeddings": 2048,
|
| 21 |
"mlp_bias": false,
|
| 22 |
"model_type": "llama",
|
model.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:fd773e3f119e9b85072142c0ad93da1fdd4a15250d662570e126cb877a93d712
|
| 3 |
+
size 538092792
|
training_args.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 5368
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e0664fe9651d5f287e72f15a82ec5653e84e0ea65404dfa3ac0a803f42ce6443
|
| 3 |
size 5368
|