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",
"GRU",
"RNN",
"Conv3d"
]
UNMANAGED_MODULES_INCLUDES = ["RotaryEmbedding", "Norm"]