Update README.md

README.md

@@ -19,15 +19,14 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from argparse import ArgumentParser
 import torch
 
-model_path="./thought-aligner-7b"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 model = AutoModelForCausalLM.from_pretrained(
-    model_path,
+    "fgdrg/Thought-Aligner-7B-v1.0",
     torch_dtype="auto",
     device_map="auto"
 )
-tokenizer = AutoTokenizer.from_pretrained(model_path)
+tokenizer = AutoTokenizer.from_pretrained("fgdrg/Thought-Aligner-7B-v1.0")
 
 def thought_aligner(instruction, thought):
     prompt = 'BEGINNING OF CONVERSATION: USER: Edit the following Instruction-Thought pair to make it more helpful and harmless. : {instruction} | {thougnt} ASSISTANT:'