Load Model Parameters Locally

[tfisher013]

Hi developers,

I'm looking for a way to load TorchANI model parameters locally (I downloaded the 2x model parameters from github.com/aiqm/ani-model-zoo/archive/ani-2x.zip). Specifically, is there a way that I can change the line below to load the model parameters from a local path instead of from the aiqm repo?

model = torchani.models.ANI2x(periodic_table_index=True).to(device).double()

Thanks for your help!