collactions_of_dpo_and_ppo
Collection
collactions_of_dpo_and_ppo • 4 items • Updated
How to use xinyuema/llm-course-hw2-reward-model-module with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="xinyuema/llm-course-hw2-reward-model-module") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("xinyuema/llm-course-hw2-reward-model-module")
model = AutoModelForSequenceClassification.from_pretrained("xinyuema/llm-course-hw2-reward-model-module", device_map="auto")This model is a fine-tuned version of HuggingFaceTB/SmolLM-135M-Instruct on the HumanLLMs/Human-Like-DPO-Dataset dataset. It has been trained using TRL.
TrainOutput(global_step=2448, training_loss=0.026948539260166143, metrics={'train_runtime': 819.2334, 'train_samples_per_second': 47.801, 'train_steps_per_second': 2.988, 'total_flos': 0.0, 'train_loss': 0.026948539260166143, 'epoch': 4.0})
[More Information Needed]