File size: 454 Bytes
ff2b095 dbb0d76 d71bc58 726a2f5 d71bc58 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
---
library_name: transformers
tags: []
---
## Model
* Base Model: [google/gemma-2-9b-it](https://huggingface.co/google/gemma-2-9b-it)
## Dataset
* custom DPO dataset
## For use
#### Load model directly
``` python
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("ENERGY-DRINK-LOVE/gemma-2-9b_dpo-v4-hq")
model = AutoModelForCausalLM.from_pretrained("ENERGY-DRINK-LOVE/gemma-2-9b_dpo-v4-hq")
``` |