more changes

.github/workflows/_demo-test.yml

@@ -13,7 +13,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        modules: [torch_to_jax]
+        modules: [image_segmentation_with_ivy_unet] #[alexnet_demo, bert_demo, image_segmentation_with_ivy_unet, mmpretrain_to_jax, resnet_demo, torch_to_jax]
 
     steps:
         - name: Clean repository
@@ -43,4 +43,3 @@ jobs:
           run: |
             docker pull unifyai/ivy:latest-gpu
             docker run --rm --gpus all -v "$(pwd)"/demos:/ivy/demos -v "$(pwd)"/ivy:/ivy/ivy unifyai/ivy:latest-gpu demos/tests/test_demos.sh ${{ secrets.USER_API_KEY }} examples_and_demos/${{ matrix.modules }}.ipynb ${{ inputs.nightly_binaries }}
-            docker rmi unifyai/ivy:latest-gpu