Skip to content

Unknown Torch dependence during get_model? #322

Description

@ensol-JKY

I'm raising this issue just for your information.

Runtime error occurs during LAMMPS simulation using LAMMPS-torch when I deploy sevennet checkpoint using get_model cli under following environment

-python 3.10
-torch 2.6.1
-cuda 12.9
-flash_TP 0.1.0

and the summarized error message is,

Traceback of Torchscript, serialized code (most recent call last):
File "code/torch/sevenn/nn/sequential/___torch_mangle_194.py" line 20, in forward
...

Traceback of Torchscript, original code (most recent call last):
File "/my/home/path/miniconda3/envs/sevennetenv/lib/python3.10/site-packages/sevenn/nn/sequential.py", line 167, in forward
..
RuntimeError: KeyeError: edge_vec

This bug happen only when I try to deploy sevenn model with sevenn version 0.13.
Is there any unknown dependence?

Deprecation of torch.jit.script, torch,jit,save, torch,trace_method may be one reason considering that there are DeprecationWarnings during deploying sevenn model using sevenn version 0.12

Or integration of ReEWC made some errors?

I have no idea, but I hope this information to help your maintenance.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions