diff --git a/openmmapi/src/TorchForceImpl.cpp b/openmmapi/src/TorchForceImpl.cpp index 054bc846..61c9b5be 100644 --- a/openmmapi/src/TorchForceImpl.cpp +++ b/openmmapi/src/TorchForceImpl.cpp @@ -46,7 +46,7 @@ TorchForceImpl::~TorchForceImpl() { } void TorchForceImpl::initialize(ContextImpl& context) { - auto module = owner.getModule(); + auto module = owner.getModule().clone(); // Create the kernel. kernel = context.getPlatform().createKernel(CalcTorchForceKernel::Name(), context); kernel.getAs().initialize(context.getSystem(), owner, module);