adding nightly flag back

pytorch · Jun 6, 2024 · 294dcb8 · 294dcb8
1 parent 5895e0b
commit 294dcb8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/benchmark_torch_compile_nightly.yml b/.github/workflows/benchmark_torch_compile_nightly.yml
@@ -45,4 +45,4 @@ jobs:
           chmod +x benchmarks/benchmark_model_dependencies.sh
           source benchmarks/benchmark_model_dependencies.sh
       - name: Benchmark gpu nightly
-        run: python benchmarks/auto_benchmark.py --input benchmarks/benchmark_config_torch_compile_gpu.yaml --skip false #--nightly True
+        run: python benchmarks/auto_benchmark.py --input benchmarks/benchmark_config_torch_compile_gpu.yaml --skip false --nightly True
diff --git a/ts_scripts/install_dependencies.py b/ts_scripts/install_dependencies.py
@@ -141,7 +141,7 @@ def install_python_packages(self, cuda_version, requirements_file_path, nightly)
                 f"pip3 install numpy --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/{pt_nightly}"
             )
             os.system(
-                f"pip3 install --pre torchtext --index-url https://download.pytorch.org/whl/nightly/cpu"
+                f"pip3 install --pre torchtext --index-url https://download.pytorch.org/whl/nightly/cpu --no-deps"
             )
         else:
             self.install_torch_packages(cuda_version)