camlab-ethz/poseidon

scOT.models import not working

visujadhav opened this issue · 2 comments

After installation from git, and running following code

`from scOT.models import ScOT

model = ScOT.from_pretrained("camlab-ethz/Poseidon-<MODEL_SIZE>")`

I am getting error as follows:
ModuleNotFoundError: No module named 'scOT.models'

Please advise about the potential solution for this case.

Just found the solution:
replaced from scOT.models import ScOT
with from scOT.model import ScOT

Request authors to update the README file.

Thanks a lot for pointing this out. I just fixed it in the README.