Update README.md
Browse files
README.md
CHANGED
|
@@ -55,14 +55,7 @@ encoding = tokenizer(ytb['Title'][i], return_tensors="pt")
|
|
| 55 |
|
| 56 |
input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
|
| 57 |
|
| 58 |
-
outputs = model.generate(
|
| 59 |
-
|
| 60 |
-
input_ids=input_ids, attention_mask=attention_masks,
|
| 61 |
-
|
| 62 |
-
max_length=30,
|
| 63 |
-
|
| 64 |
-
early_stopping=True
|
| 65 |
-
)
|
| 66 |
|
| 67 |
for output in outputs:
|
| 68 |
|
|
|
|
| 55 |
|
| 56 |
input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
|
| 57 |
|
| 58 |
+
outputs = model.generate(input_ids=input_ids, attention_mask=attention_masks,max_length=30,early_stopping=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
|
| 60 |
for output in outputs:
|
| 61 |
|