🐛 [BUG] Segfault with float64 models about nequip HOT 4 CLOSED

In [1]: from nequip.ase import NequIPCalculator

In [3]: calc = NequIPCalculator.from_deployed_model("test.pth", "cuda")
/home/a/Documents/NequIP/code/nequip/nequip/utils/_global_options.py:45: UserWarning: Setting the GLOBAL value for allow_tf32 to False which is different than the previous value of True
  warnings.warn(
/home/a/Documents/NequIP/code/nequip/nequip/utils/_global_options.py:58: UserWarning: Setting the GLOBAL value for jit fusion strategy to `[('DYNAMIC', 3)]` which is different than the previous value of `[('STATIC', 2), ('DYNAMIC', 10)]`
  warnings.warn(
/home/a/Documents/NequIP/code/nequip/nequip/ase/nequip_calculator.py:73: UserWarning: Trying to use chemical symbols as NequIP type names; this may not be correct for your model! To avoid this warning, please provide `species_to_type_name` explicitly.
  warnings.warn(

In [4]: import ase.build

In [6]: atoms = ase.build.molecule("CH3CHO")

In [7]: atoms.calc = calc

In [8]: atoms.get_potential_energy()
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
...
RuntimeError: The following operation failed in the TorchScript interpreter.
Traceback of TorchScript, serialized code (most recent call last):
...
RuntimeError: expected scalar type Float but found Double


In [9]: import torch

In [10]: torch.set_default_dtype(torch.float64)

In [12]: atoms.get_potential_energy()
Out[12]: -135216.08807821615