jingyeom's picture
Update README.md
726a2f5 verified
|
raw
history blame
No virus
454 Bytes
---
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")
```