finex commited on
Commit
f8ccaa4
·
verified ·
1 Parent(s): 5eef48e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -41,7 +41,7 @@ tokenizer = AutoTokenizer.from_pretrained("finex/Stage-IOTGraphic")
41
  model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
42
 
43
  # Let's chat for 5 lines
44
- #for step in range(5):
45
  # encode the new user input, add the eos_token and return a tensor in Pytorch
46
  new_user_input_ids = tokenizer.encode(input(">> User:") + tokenizer.eos_token, return_tensors='pt')
47
 
 
41
  model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
42
 
43
  # Let's chat for 5 lines
44
+ #for step in range(5):
45
  # encode the new user input, add the eos_token and return a tensor in Pytorch
46
  new_user_input_ids = tokenizer.encode(input(">> User:") + tokenizer.eos_token, return_tensors='pt')
47