Skip to content

Exporting models without saving them on disk#89

Open
abmazitov wants to merge 3 commits intomainfrom
fix-export
Open

Exporting models without saving them on disk#89
abmazitov wants to merge 3 commits intomainfrom
fix-export

Conversation

@abmazitov
Copy link
Collaborator

@abmazitov abmazitov commented Feb 4, 2026

As mentioned in #88 , in the current implementation of the UPETCalculator model undergoes torch-scripting by calling the AtomisticModel.save() method, that requires saving a temporary model file on disk. This can be avoided by directly scripting the model in get_upet() function and initialising the calculator with a scripted version.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant