lorax
lorax copied to clipboard
AutoTokenzier.from_pretrains needs setting with `trust_remote_code` inside `load_module_map`
System Info
lorax main
Information
- [X] Docker
- [ ] The CLI directly
Tasks
- [ ] An officially supported command
- [ ] My own modifications
Reproduction
- using a tokenizer class not registered in the Transformers
- place the tokenizer config file and related files inside the adapter checkpoint folder
- inference with adapter info pointed to this adapter folder
- the tokenizer loading will go into the resolve_trust_remote_code logic inside the Transformers, where it will require user input to confirm if
trust_remote_codeis not True.
Expected behavior
loading the tokenizer directly if it is existed inside adapter.
The fix is easy, will prepare a PR for review soon.
Thanks @thincal!