TE

Signed-off-by: oliver könig <[email protected]>
NVIDIA · Jan 19, 2025 · 4311ab5 · 4311ab5
1 parent c12853e
commit 4311ab5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.ci b/Dockerfile.ci
@@ -52,7 +52,7 @@ ARG NEMO_RUN_TAG=34259bd3e752fef94045a9a019e4aaf62bd11ce2
 RUN pip install nemo_run@git+https://github.com/NVIDIA/NeMo-Run.git@${NEMO_RUN_TAG}
 
 # Install NeMo requirements
-ARG TE_TAG=7d576ed25266a17a7b651f2c12e8498f67e0baea
+ARG TE_TAG=2215fa5c7557b66034068816020f9f611019e457
 ARG MODELOPT_VERSION=0.21.0
 ARG MCORE_TAG=bd677bfb13ac2f19deaa927adc6da6f9201d66aa