search

lightonai / RITA

RITA is a family of autoregressive protein models, developed by LightOn in collaboration with the OATML group at Oxford and the Debora Marks Lab at Harvard.
MIT License
88 stars 8 forks source link

cannot run example #5

Closed lucidrains closed 2 years ago

lucidrains commented 2 years ago

Hi,

I'm trying to run your example, but seeing

[/root/.cache/huggingface/modules/transformers_modules/lightonai/RITA_s/rita_modeling_ad8dfe2e2240d5a6abcfa9ea52f3868a35f07f4670c887c31d420f6bba0fdc5f_3db732a66a256d53b183fe3a079bbfbf2db700c63f7789178c37924666055f28.py](https://localhost:8080/#) in <module>()
     18 from transformers.utils import logging
     19 
---> 20 from .rita_configuration import RITAConfig
     21 import torch.nn.functional as F
     22 logger = logging.get_logger(__name__)

ModuleNotFoundError: No module named 'transformers_modules.lightonai.RITA_s.rita_configuration'
DanielHesslow commented 2 years ago

Hrm, I cannot reproduce this issue

Here's a colab running the example: https://colab.research.google.com/drive/16GO9fhJwCjcrFBL3MrGHsa9fyBAgth2n?usp=sharing

Maybe try upgrading transformers and clearing the cache?

lucidrains commented 2 years ago

@DanielHesslow oh yes, all good after upgrading transformers to 4.15, thanks Daniel!