Upload README.md
Browse files
README.md
CHANGED
|
@@ -93,7 +93,6 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
|
|
| 93 |
# Define your prompt
|
| 94 |
prompt = "The curious cat looked at the"
|
| 95 |
|
| 96 |
-
# IMPORTANT: Use tokenizer without special tokens
|
| 97 |
inputs = tokenizer(prompt, return_tensors="pt")
|
| 98 |
input_ids = inputs.input_ids.to("cuda")
|
| 99 |
|
|
|
|
| 93 |
# Define your prompt
|
| 94 |
prompt = "The curious cat looked at the"
|
| 95 |
|
|
|
|
| 96 |
inputs = tokenizer(prompt, return_tensors="pt")
|
| 97 |
input_ids = inputs.input_ids.to("cuda")
|
| 98 |
|