Model save

README.md

@@ -0,0 +1,75 @@
+---
+library_name: transformers
+license: gemma
+base_model: google/gemma-2-9b-it
+tags:
+- trl
+- dpo
+- llama-factory
+- generated_from_trainer
+model-index:
+- name: gemma-oh-preferences
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# gemma-oh-preferences
+
+This model is a fine-tuned version of [google/gemma-2-9b-it](https://huggingface.co/google/gemma-2-9b-it) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 3.9479
+- Rewards/chosen: -46.2475
+- Rewards/rejected: -47.8884
+- Rewards/accuracies: 0.6453
+- Rewards/margins: 1.6409
+- Logps/rejected: -4.7888
+- Logps/chosen: -4.6247
+- Logits/rejected: 14.7121
+- Logits/chosen: 14.4778
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 8e-07
+- train_batch_size: 2
+- eval_batch_size: 2
+- seed: 42
+- distributed_type: multi-GPU
+- num_devices: 32
+- gradient_accumulation_steps: 2
+- total_train_batch_size: 128
+- total_eval_batch_size: 64
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 1.0
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rewards/chosen | Rewards/rejected | Rewards/accuracies | Rewards/margins | Logps/rejected | Logps/chosen | Logits/rejected | Logits/chosen |
+|:-------------:|:-----:|:----:|:---------------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|
+| 3.9886        | 1.0   | 7344 | 3.9479          | -46.2475       | -47.8884         | 0.6453             | 1.6409          | -4.7888        | -4.6247      | 14.7121         | 14.4778       |
+
+
+### Framework versions
+
+- Transformers 4.45.2
+- Pytorch 2.3.0
+- Datasets 2.21.0
+- Tokenizers 0.20.2

generation_config.json

@@ -0,0 +1,8 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 2,
+  "cache_implementation": "hybrid",
+  "eos_token_id": 1,
+  "pad_token_id": 0,
+  "transformers_version": "4.45.2"
+}

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:10e697d43b8906e9923033d78b83d911091b09918ba4b32b63cbffb84dea564c
-size 7160
+oid sha256:7b2f45b26c05c38a710894d838cbcea30e718bbeabfe4bcb34e06441a0f31e37
+size 7096