Two new options: truncation length and ban eos token
This commit is contained in:
parent
749c08a4ff
commit
cacbcda208
6 changed files with 62 additions and 48 deletions
|
@ -189,7 +189,6 @@ def load_model(model_name):
|
|||
pass
|
||||
else:
|
||||
tokenizer = AutoTokenizer.from_pretrained(Path(f"{shared.args.model_dir}/{shared.model_name}/"))
|
||||
tokenizer.truncation_side = 'left'
|
||||
|
||||
print(f"Loaded the model in {(time.time()-t0):.2f} seconds.")
|
||||
return model, tokenizer
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue