Move PyTorch nightly installation from runtime to build requirements.txt

app.py

@@ -11,11 +11,6 @@ print("=" * 50, flush=True)
 print("APP.PY STARTING - DEBUG VERSION", flush=True)
 print("=" * 50, flush=True)
 
-# Install nightly versions for FA3 + FP8 support
-print("Installing PyTorch nightly...", flush=True)
-exit_code = os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 torch torchvision torchao spaces')
-print(f"pip install exit code: {exit_code}", flush=True)
-
 import logging
 from datetime import datetime
 

requirements.txt

@@ -1,7 +1,9 @@
-# Core PyTorch dependencies (installed at runtime in app.py for CUDA 12.6 compatibility)
-# torch==2.8.0 (from https://download.pytorch.org/whl/nightly/cu126)
-# torchvision==0.23.0 (from https://download.pytorch.org/whl/nightly/cu126)
-# torchao==0.16.0.dev (from https://download.pytorch.org/whl/nightly/cu126)
+--extra-index-url https://download.pytorch.org/whl/nightly/cu126
+
+# Core PyTorch dependencies (installed during build)
+torch --pre
+torchvision --pre
+torchao --pre
 
 # HuggingFace ecosystem (pinned versions for compatibility)
 transformers>=4.40,<5.0