Update smoke_test.py

.ci/pytorch/smoke_test/smoke_test.py

@@ -33,6 +33,7 @@ MODULES = [
         "smoke_test": "./vision/test/smoke_test.py",
         "extension": "extension",
         "repo_name": "vision",
+        "branch": "release/0.21",
     },
     {
         "name": "torchaudio",
@@ -40,6 +41,7 @@ MODULES = [
         "smoke_test": "./audio/test/smoke_test/smoke_test.py --no-ffmpeg",
         "extension": "_extension",
         "repo_name": "audio",
+        "branch": "release/2.6",
     },
 ]
 
@@ -313,7 +315,7 @@ def smoke_test_modules():
                 print(f"Path does not exist: {cwd}/{module['repo_name']}")
                 try:
                     subprocess.check_output(
-                        f"git clone --depth 1 {module['repo']}",
+                        f"git clone --depth 1 -b {module['branch']} {module['repo']}",
                         stderr=subprocess.STDOUT,
                         shell=True,
                     )