polieste commited on
Commit
a0b942b
·
1 Parent(s): 42d595a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -28,7 +28,7 @@ encoding = tokenizer(text, return_tensors="pt")
28
  input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
29
  outputs = model.generate(
30
  input_ids=input_ids, attention_mask=attention_masks,
31
- max_length=128,
32
  early_stopping=True
33
  )
34
  for output in outputs:
 
28
  input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
29
  outputs = model.generate(
30
  input_ids=input_ids, attention_mask=attention_masks,
31
+ max_length=512,
32
  early_stopping=True
33
  )
34
  for output in outputs: