Update README.md

README.md

@@ -56,7 +56,7 @@ inputs = tokenizer.apply_chat_template({
   "copyright": "original",
   "character": "",
   "general": "1girl"
-}, tokenize=True) # tokenize=False to preview prompt
+}, return_tensors="pt", tokenize=True) # tokenize=False to preview prompt
 # same as input_ids of "<|bos|><rating>rating:sfw, rating:general</rating><copyright>original</copyright><character></character><general>1girl"
 
 with torch.no_grad():
@@ -108,7 +108,7 @@ inputs = tokenizer.apply_chat_template({
   "copyright": "original",
   "character": "",
   "general": "1girl"
-}, tokenize=True)
+}, return_tensors="pt", tokenize=True,)
 
 with torch.no_grad():
   outputs = ort_model.generate(inputs, generation_config=generation_config)