From 4486170ff171d0f640d591409ebc4011233aaf07 Mon Sep 17 00:00:00 2001
From: LeiZhang <48861350+leoz9@users.noreply.github.com>
Date: Thu, 14 Nov 2024 19:47:26 +0800
Subject: [PATCH] Update 5-dpo_train.py

Modify the huggingface warehouse address to facilitate direct retrieval
---
 5-dpo_train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/5-dpo_train.py b/5-dpo_train.py
index ef17e2c..2179c01 100644
--- a/5-dpo_train.py
+++ b/5-dpo_train.py
@@ -12,8 +12,8 @@
 def init_model():
     device = 'cuda:0'
     # Do model patching and add fast LoRA weights
-    model_name_or_path = "minimind-v1"
-    tokenizer_name_or_path = "minimind-v1"
+    model_name_or_path = "jingyaogong/minimind-v1"
+    tokenizer_name_or_path = "jingyaogong/minimind-v1"
     model = AutoModelForCausalLM.from_pretrained(model_name_or_path, trust_remote_code=True)
     tokenizer = AutoTokenizer.from_pretrained(tokenizer_name_or_path, trust_remote_code=True, use_fast=False)
     tokenizer.pad_token = tokenizer.eos_token