rs239 / ablm

Protein language model customized for antibodies
MIT License
60 stars 8 forks source link

_pickle.UnpicklingError: invalid load key, '<'. #5

Closed xinycui closed 1 year ago

xinycui commented 1 year ago

Hi, when I submitted demo_detailed.py, the following problem occurred, can you help me solve it.

image
lordim commented 1 year ago

Hello, I cannot seem to replicate your error when I use reload_models_to_device(0). I think I need a bit more context as to how you used that function in your python script. Could you send me the relevant lines in your demo_detailed.py?