search

mheinzinger / ProstT5

Bilingual Language Model for Protein Sequence and Structure
MIT License
183 stars 15 forks source link

Readme update: model.full() - AttributeError: 'T5EncoderModel' object has no attribute 'full' #35

Open robsyc opened 1 week ago

robsyc commented 1 week ago

For people trying to get the model running locally on CPU, it might be beneficial to update the instructions in README.md to:

model.float() if device.type=='cpu' else model.half()

Rather than

model.full() if device=='cpu' else model.half()

The latter causes an exception