ModuleNotFoundError: No module named 'transformers.models.opt'

[MaximeTut]

On windows 10, python 3.9, I pip install galai without error but I got this error when importing galai:

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
~\AppData\Local\Temp\ipykernel_640\446023240.py in <module>
----> 1 import galai

~\anaconda3\lib\site-packages\galai\__init__.py in <module>
----> 1 from galai.model import Model
      2 from galai.utils import get_checkpoint_path, get_tokenizer_path
      3 
      4 
      5 def load_model(name: str, dtype: str=None, num_gpus: int=None):

~\anaconda3\lib\site-packages\galai\model.py in <module>
      6 from tokenizers import Tokenizer
      7 
----> 8 from galai.architecture import OPTForCausalLM, OPTConfig
      9 from galai.utils import escape_custom_split_sequence
     10 

~\anaconda3\lib\site-packages\galai\architecture.py in <module>
     11 
     12 from galai.config import OPTConfig
---> 13 from transformers.models.opt.modeling_opt import (
     14     ACT2FN,
     15     BaseModelOutputWithPast,

ModuleNotFoundError: No module named 'transformers.models.opt'

[brandonbiggs]

This is usually solved by running pip install transformers.

[kmarz84]

In my case, the problem was solved by updating transformers.

[mkardas]

Hi @MaximeTut, thanks for reporting. We added a minimum transformers version to the dependencies, so it should work now.