jingyeom's picture
Update README.md
726a2f5 verified
|
raw
history blame
No virus
454 Bytes
metadata
library_name: transformers
tags: []

Model

Dataset

  • custom DPO dataset

For use

Load model directly

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")