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

How to make text-gen use a newer git version of huggingface/transformers? #87

Open
futu2 opened this issue Mar 2, 2024 · 1 comment

Comments

@futu2
Copy link

futu2 commented Mar 2, 2024

Since some newer models like starcoder2 are not included in model config and when load model it would complain:

  File "/nix/store/7hpffz24mjm12y5ymd2is43lxl7nf27b-python3-3.11.6-env/lib/python3.11/site-packages/transformers/models/auto/configuration_auto.py", line 1064, in from_pretrained
    config_class = CONFIG_MAPPING[config_dict["model_type"]]
                   ~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/nix/store/7hpffz24mjm12y5ymd2is43lxl7nf27b-python3-3.11.6-env/lib/python3.11/site-packages/transformers/models/auto/configuration_auto.py", line 761, in __getitem__
    raise KeyError(key)
KeyError: 'starcoder2'

In normal env, pip install --upgrade git+https://github.com/huggingface/transformers would work.

@max-privatevoid
Copy link
Member

Right now we're using transformers from nixpkgs, which isn't a new enough version to support starcoder2 yet. We could package our own version of transformers.

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

No branches or pull requests

2 participants