From 7abf5e20be21868ed18a8067db8a12aa086ff1e8 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Wed, 15 Oct 2025 10:12:06 -0600 Subject: [PATCH] Add conv3d to memory management excluded modules --- toolkit/memory_management/manager.py | 1 + 1 file changed, 1 insertion(+) 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"]