evan/ai-toolkit
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add conv3d to memory management excluded modules

Browse Source
This commit is contained in:
Jaret Burkett committed 2025-10-15 10:12:06 -06:00
1 parent 91b87e06a1
commit 7abf5e20be
1 file changed
+1
Download Patch File Download Diff File Expand all files Collapse all files
toolkit/memory_management/manager.py
+1
View File
@@ -28,6 +28,7 @@ UNMANAGED_MODULES = [
"LSTM", "LSTM",
"GRU", "GRU",
"RNN", "RNN",
"Conv3d"
] ]
UNMANAGED_MODULES_INCLUDES = ["RotaryEmbedding", "Norm"] UNMANAGED_MODULES_INCLUDES = ["RotaryEmbedding", "Norm"]