Install entire TE

Rohan-Bierneni · Rohan-Bierneni · commit 9236f6fcfd6e · 2025-09-26T17:25:38.000Z
diff --git a/.github/workflows/UnitTests.yml b/.github/workflows/UnitTests.yml
@@ -37,6 +37,8 @@ jobs:
 
       - name: Print dependencies
         run: |
+          pip uninstall -y transformer-engine
+          pip install transformer-engine
           pip freeze
 
       - name: Run MaxDiffusion Training
