Update UnitTests.yml

Author: parambole

.github/workflows/UnitTests.yml

@@ -58,4 +58,5 @@ jobs:
             output_dir="/tmp/flux-output/" \
             max_train_steps=5 \
             hardware=gpu \
-            attention="cudnn_flash_te"
+            attention="cudnn_flash_te" \
+            enable_checkpointing=False \