Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

AutoTokenzier.from_pretrains needs setting with trust_remote_code inside load_module_map #466

Closed
1 of 4 tasks
thincal opened this issue May 13, 2024 · 2 comments · Fixed by #476
Closed
1 of 4 tasks
Labels
enhancement New feature or request

Comments

@thincal
Copy link
Contributor

thincal commented May 13, 2024

System Info

lorax main

Information

  • Docker
  • The CLI directly

Tasks

  • An officially supported command
  • My own modifications

Reproduction

  1. using a tokenizer class not registered in the Transformers
  2. place the tokenizer config file and related files inside the adapter checkpoint folder
  3. inference with adapter info pointed to this adapter folder
  4. 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_code is not True.

Expected behavior

loading the tokenizer directly if it is existed inside adapter.

@thincal
Copy link
Contributor Author

thincal commented May 13, 2024

The fix is easy, will prepare a PR for review soon.

@tgaddair
Copy link
Contributor

Thanks @thincal!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants