diff --git a/toolkit/memory_management/manager.py b/toolkit/memory_management/manager.py index dbb0ab8d..59b63806 100644 --- a/toolkit/memory_management/manager.py +++ b/toolkit/memory_management/manager.py @@ -28,6 +28,7 @@ UNMANAGED_MODULES = [ "LSTM", "GRU", "RNN", + "Conv3d" ] UNMANAGED_MODULES_INCLUDES = ["RotaryEmbedding", "Norm"]