search

chaidiscovery / chai-lab

Chai-1, SOTA model for biomolecular structure prediction
https://www.chaidiscovery.com
Apache License 2.0
1.35k stars 162 forks source link

error: AttributeError: module 'torch.library' has no attribute 'register_fake' #138

Open lhallee opened 4 weeks ago

lhallee commented 4 weeks ago

Hello,

I had to install torchvision==0.18.1 because of a compatibility with pytorch 2.3.1 and transformers 4.44, otherwise I would get this error

AttributeError: module 'torch.library' has no attribute 'register_fake'

May need to add to requirements. Cheers, Logan

arogozhnikov commented 4 weeks ago

Hi Logan, thanks for reporting.

Current setup (without torchvision) is for sure working, the problem is likely introduced by a new version of torchvision: https://github.com/huggingface/diffusers/issues/8958

lhallee commented 4 weeks ago

Looks like the issue had to do with my Azure instance holding a transformers 4.46 version even though I did a --force-reinstall --no-cache-dir... All set now, thanks

arogozhnikov commented 1 week ago

reopening to keep this visible for other people bumping into the same issue