diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml
index 9dc2ca2eb2469..a2d495138e8c3 100644
--- a/.github/workflows/testing.yml
+++ b/.github/workflows/testing.yml
@@ -250,7 +250,7 @@ jobs:
             /home/dev/taichi/.github/workflows/scripts/aot-demo.sh build-and-test-headless-demo-desktop
         env:
           PY: '3.8'
-          TI_DEVICE_MEMORY_GB: '1'
+          TI_DEVICE_MEMORY_GB: '0.1'
 
       - name: Save wheel if test failed
         if: failure() && steps.test.conclusion == 'failure'
@@ -580,3 +580,5 @@ jobs:
           ci-docker-run-gpu --name taichi-test-capi-headless-demo \
             registry.taichigraphics.com/taichidev-androidsdk18.04:v0.0.6 \
             /home/dev/taichi/.github/workflows/scripts/aot-demo.sh build-and-test-headless-demo
+        env:
+          TI_DEVICE_MEMORY_GB: '0.1'