lucidrains/enformer-pytorch

How to load the pre-trained Enfromer model?

yzJiang9 opened this issue · 2 comments

yzJiang9 commented

Hi, I encountered a problem when trying to load the pre-trained enformer model.

from enformer_pytorch import Enformer
model = Enformer.from_pretrained("EleutherAI/enformer-preview")

AttributeError Traceback (most recent call last)
Input In [3], in
1 from enformer_pytorch import Enformer
----> 2 model = Enformer.from_pretrained("EleutherAI/enformer-preview")

AttributeError: type object 'Enformer' has no attribute 'from_pretrained'

lucidrains commented

@yzJiang9 hey! if you upgrade to version 0.4, it will be fixed

pip install enformer-pytorch --upgrade

  • 👍1
yzJiang9 commented

I will definitely give it a try... I installed the package like one month ago.

  • 👍1