Update README.md
Browse files
README.md
CHANGED
@@ -39,7 +39,8 @@ prompts = [
|
|
39 |
stampa 'FizzBuzz'."
|
40 |
]
|
41 |
|
42 |
-
inputs = tokenizer(prompts, return_tensors="pt", padding=True,
|
|
|
43 |
|
44 |
with torch.no_grad():
|
45 |
gen_outputs = model.generate(
|
|
|
39 |
stampa 'FizzBuzz'."
|
40 |
]
|
41 |
|
42 |
+
inputs = tokenizer(prompts, return_tensors="pt", padding=True, \
|
43 |
+
truncation=True).to(model.device)
|
44 |
|
45 |
with torch.no_grad():
|
46 |
gen_outputs = model.generate(
|