Update README.md
Browse files
README.md
CHANGED
|
@@ -60,7 +60,7 @@ inputs = tokenizer(
|
|
| 60 |
max_length=1280,
|
| 61 |
truncation=True,
|
| 62 |
return_tensors="pt"
|
| 63 |
-
)
|
| 64 |
|
| 65 |
outputs = model.generate(
|
| 66 |
inputs.input_ids,
|
|
|
|
| 60 |
max_length=1280,
|
| 61 |
truncation=True,
|
| 62 |
return_tensors="pt"
|
| 63 |
+
).to("cuda")
|
| 64 |
|
| 65 |
outputs = model.generate(
|
| 66 |
inputs.input_ids,
|