johnbridges commited on
Commit
e29d9b0
·
1 Parent(s): c4a1db4
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -18,11 +18,12 @@ huggingface_hub
18
  vllm>=0.10.0
19
  transformers>=4.51.0
20
 
 
21
  torch==2.8.0
22
  intel-extension-for-pytorch==2.8.0
23
 
24
- # TimesFM from source + deps (replaces plain 'timesfm')
25
- git+https://github.com/google-research/timesfm.git@main
26
  numpy==1.26.4
27
  einops>=0.7
28
  pandas>=2.0,<2.3
 
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
24
 
25
+ # TimesFM from source + deps
26
+ git+https://github.com/google-research/timesfm.git@master
27
  numpy==1.26.4
28
  einops>=0.7
29
  pandas>=2.0,<2.3