Removing references to specific branch of test-infra and testing with @main

Author: jeanschmidt

.github/workflows/float8_test.yml

@@ -10,9 +10,9 @@ on:
       - main
       - 'gh/**'
 
-# concurrency:
-#   group: float8_test-${{ github.workflow }}-${{ github.ref == 'refs/heads/main' && github.run_number || github.ref }}
-#   cancel-in-progress: true
+concurrency:
+  group: float8_test-${{ github.workflow }}-${{ github.ref == 'refs/heads/main' && github.run_number || github.ref }}
+  cancel-in-progress: true
 
 env:
   HF_TOKEN: ${{ secrets.HF_TOKEN }}
@@ -29,13 +29,12 @@ jobs:
             gpu-arch-type: "cuda"
             gpu-arch-version: "12.1"
 
-    uses: pytorch/test-infra/.github/workflows/linux_job.yml@jeanschmidt/linux_job_containers
+    uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       timeout: 60
       runner: ${{ matrix.runs-on }}
       gpu-arch-type: ${{ matrix.gpu-arch-type }}
       gpu-arch-version: ${{ matrix.gpu-arch-version }}
-      test-infra-ref: jeanschmidt/linux_job_containers
       script: |
         conda create -n venv python=3.9 -y
         conda activate venv