BAAI
/

Gemma2-9B-IT-Simpo-Infinity-Preference

Model card Files Files and versions Community

hyxmmm commited on Sep 1, 2024

Commit

028a91b

·

verified ·

1 Parent(s): 139fb26

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ model = AutoModelForCausalLM.from_pretrained("BAAI/Gemma2-9B-IT-Simpo-Infinity-P
     torch_dtype=torch.bfloat16,
     device_map="auto"
 )
-tokenizer = AutoTokenizer.from_pretrained("BAAI/Infinity-Instruct-3M-0625-Llama3-70B")
 prompt = "Give me a short introduction to large language model."
 messages = [

     torch_dtype=torch.bfloat16,
     device_map="auto"
 )
+tokenizer = AutoTokenizer.from_pretrained("BAAI/Gemma2-9B-IT-Simpo-Infinity-Preference")
 prompt = "Give me a short introduction to large language model."
 messages = [