Update README.md

README.md

@@ -57,7 +57,7 @@ inputs = tokenizer.apply_chat_template({
   "character": "",
   "general": "1girl",
   "length": "<|long|>"
-}, 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><|long|>1girl<|input_end|>"
 with torch.no_grad():
   outputs = model.generate(inputs, generation_config=generation_config)
@@ -109,7 +109,7 @@ inputs = tokenizer.apply_chat_template({
   "character": "",
   "general": "1girl",
   "length": "<|long|>"
-}, tokenize=True)
+}, return_tensors="pt", tokenize=True)
 
 with torch.no_grad():
   outputs = ort_model.generate(inputs, generation_config=generation_config)