Streamlit cannot hash pytorch model

Hi guys - keep getting this error:

InternalHashError : While caching the arguments of build_query() , Streamlit encountered an object of type torch.nn.backends.thnn.THNNFunctionBackend , which it does not know how to hash.

Request some help! Thanks