run ut based on tf5.2.0

Author: kkscilife

.github/workflows/unit_test.yaml

@@ -1,5 +1,6 @@
 name: unit_test
 on:
+  push:
   pull_request:
     branches:
       - "main"
@@ -14,7 +15,7 @@ on:
 env:
   WORKSPACE_PREFIX: $(echo $GITHUB_WORKSPACE |cut -d '/' -f 1-5)
   WORKSPACE_PREFIX_SHORT: $(echo $GITHUB_WORKSPACE |cut -d '/' -f 1-3)
-  IMAGE: registry.h.pjlab.org.cn/ailab-llmrazor/xtuner:pt28_20251216_d769950
+  IMAGE: registry.h.pjlab.org.cn/ailab-llmrazor/xtuner_tmp:pt28_20260303_a9c0c04
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}