Update README.md (#2)
Browse files- Update README.md (aced1c2bb5c5a09476cf292543022af44ee7b3d6)
README.md
CHANGED
|
@@ -81,7 +81,7 @@ chat_prompt = tokenizer.apply_chat_template(
|
|
| 81 |
inputs = tokenizer(chat_prompt, return_tensors="pt")
|
| 82 |
inputs.to(device)
|
| 83 |
|
| 84 |
-
with torch.inference_mode:
|
| 85 |
generate_ids = model.generate(**inputs, max_length=500)
|
| 86 |
|
| 87 |
input_len = inputs['input_ids'].shape[1]
|
|
|
|
| 81 |
inputs = tokenizer(chat_prompt, return_tensors="pt")
|
| 82 |
inputs.to(device)
|
| 83 |
|
| 84 |
+
with torch.inference_mode():
|
| 85 |
generate_ids = model.generate(**inputs, max_length=500)
|
| 86 |
|
| 87 |
input_len = inputs['input_ids'].shape[1]
|