Update README.md

README.md

@@ -37,12 +37,16 @@ MIN_TRANSFORMERS_VERSION = '4.25.1'
 assert transformers.__version__ >= MIN_TRANSFORMERS_VERSION, f'Please upgrade transformers to version {MIN_TRANSFORMERS_VERSION} or higher.'
 
 # init
-tokenizer = AutoTokenizer.from_pretrained("togethercomputer/chessgpt-chat-v1")
-model = AutoModelForCausalLM.from_pretrained("togethercomputer/chessgpt-chat-v1", torch_dtype=torch.float16)
+tokenizer = AutoTokenizer.from_pretrained("Waterhorse/chessgpt-chat-v1")
+model = AutoModelForCausalLM.from_pretrained("Waterhorse/chessgpt-chat-v1", torch_dtype=torch.float16)
 model = model.to('cuda:0')
 
 # infer
-prompt = "Alan Turing is"
+# Conversation between two
+prompt = "A friendly, helpful chat between some humans.<|endoftext|>Human 0: 1.e4 c5, what is the name of this opening?<|endoftext|>Human 1:"
+# Conversation between more than two
+#prompt = "A friendly, helpful chat between some humans.<|endoftext|>Human 0: 1.e4 c5, what is the name of this opening?<|endoftext|>Human 1: Sicilian defense.<|endoftext|>Human 2:"
+
 inputs = tokenizer(prompt, return_tensors='pt').to(model.device)
 input_length = inputs.input_ids.shape[1]
 outputs = model.generate(
@@ -51,10 +55,6 @@ outputs = model.generate(
 token = outputs.sequences[0, input_length:]
 output_str = tokenizer.decode(token)
 print(output_str)
-"""
-a name that has been synonymous with the computer age since the 1950s. The British mathematician, logician, and cryptanalyst is widely regarded as the father of modern computing. His contributions to the development of the modern computer and the theory of computation have had a profound impact on the world we live in today.
-Turing’s contributions to the development of the modern computer were made in the 1940s and 1950s. He is most famous for his work on the Turing machine, a theoretical model of a computing machine that was able to perform all the mathematical operations of a computer. Turing’s work on the...
-"""
 ```
 
 # Uses