jeanflop commited on
Commit
7239bb5
1 Parent(s): a0ff9be

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -58,7 +58,7 @@ peft_model.config.max_length=512
58
  peft_tokenizer.model_max_length=512
59
  inputs = peft_tokenizer(inputs, return_tensors="pt")
60
  outputs = peft_model.generate(**inputs,max_length=512)
61
- answer = tokenizer.decode(outputs[0])
62
  from textwrap import fill
63
 
64
  print(fill(answer, width=80))
 
58
  peft_tokenizer.model_max_length=512
59
  inputs = peft_tokenizer(inputs, return_tensors="pt")
60
  outputs = peft_model.generate(**inputs,max_length=512)
61
+ answer = peft_tokenizer.decode(outputs[0])
62
  from textwrap import fill
63
 
64
  print(fill(answer, width=80))