Update README.md
Browse files
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 |
|