johnbridges commited on
Commit
66e3763
·
1 Parent(s): 3fa43ae
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -18,6 +18,6 @@ huggingface_hub
18
  vllm>=0.10.0
19
  transformers>=4.51.0
20
 
21
- # Intel optimized PyTorch stack (CPU with BF16 oneDNN kernels)
22
- torch==2.8.0+cpu
23
  intel-extension-for-pytorch==2.8.0
 
18
  vllm>=0.10.0
19
  transformers>=4.51.0
20
 
21
+ # Intel optimized PyTorch stack (BF16-capable oneDNN build)
22
+ torch==2.8.0
23
  intel-extension-for-pytorch==2.8.0