Model save

README.md

@@ -34,7 +34,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 2e-05
+- learning_rate: 0.0001
 - train_batch_size: 8
 - eval_batch_size: 8
 - seed: 42
@@ -42,16 +42,9 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 1
 
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
-|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
-| No log        | 1.0   | 211  | 3.3125          | 0.17   | 0.026  | 0.1391 | 0.1389    | 18.7583 |
-
-
 ### Framework versions
 
 - Transformers 4.34.1
-- Pytorch 2.1.0+cu121
+- Pytorch 2.1.0
 - Datasets 2.14.6
 - Tokenizers 0.14.1

generation_config.json

@@ -1,4 +1,5 @@
 {
+  "_from_model_config": true,
   "decoder_start_token_id": 0,
   "eos_token_id": 1,
   "pad_token_id": 0,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:14f8cff59802300142b970439d2a6ef8d0c479e0f441197bf87505729e0e5a5c
-size 242072086
+oid sha256:9ff70a69c8855c5353559381a94074fffe8bc564e163955de763ab453a1dcf0a
+size 242070230

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:071c5af78761e9ac81e8f1e4c1687ac968c593d8fff08f204ee1153736c227af
-size 4664
+oid sha256:c7c1bd8dbfd786f08edd9eb83d1dca131cd16071aade49f5b1edbee457b85c30
+size 4600