s3y commited on
Commit
a6725f3
·
verified ·
1 Parent(s): ac43ad6

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +9 -7
requirements.txt CHANGED
@@ -1,23 +1,23 @@
 
 
 
1
  # Core ML/AI packages
2
  jax[cuda12]==0.5.3
3
  jaxlib==0.5.3
4
  jax-cuda12-plugin==0.5.3
5
  jax-cuda12-pjrt==0.5.3
6
- torch==2.7.1
7
- torchvision==0.22.1
8
  flax==0.10.2
9
  optax==0.2.4
10
- transformers==4.53.2
11
  diffusers==0.33.1
12
 
13
  # Computer vision and image processing
14
  opencv-python==4.11.0.86
15
- pillow>=11.0.0
16
  imageio>=2.36.1
17
 
18
  # Scientific computing (core)
19
  numpy>=1.22.4,<2.0.0
20
- scipy>=1.15.0
21
 
22
  # Data handling
23
  dm-tree>=0.1.8
@@ -41,7 +41,7 @@ beartype==0.19.0
41
  treescope>=0.1.7
42
 
43
  # File handling and storage
44
- fsspec[gcs]>=2024.6.0
45
  flatbuffers>=24.3.25
46
  h5py>=3.10.0
47
 
@@ -58,8 +58,10 @@ aiohttp>=3.8.0
58
  websockets>=11.0
59
 
60
  # Development (minimal - remove if not needed)
61
- typing-extensions>=4.12.2
 
62
 
63
  # Robotics/Control specific
64
  gymnasium>=0.29.0
65
  mujoco>=2.3.0
 
 
1
+ # Core project dependencies - minimal set to avoid system package conflicts
2
+ # Based on openpi pyproject.toml dependencies
3
+
4
  # Core ML/AI packages
5
  jax[cuda12]==0.5.3
6
  jaxlib==0.5.3
7
  jax-cuda12-plugin==0.5.3
8
  jax-cuda12-pjrt==0.5.3
 
 
9
  flax==0.10.2
10
  optax==0.2.4
 
11
  diffusers==0.33.1
12
 
13
  # Computer vision and image processing
14
  opencv-python==4.11.0.86
15
+ # pillow>=11.0.0 # Removed - system package conflict
16
  imageio>=2.36.1
17
 
18
  # Scientific computing (core)
19
  numpy>=1.22.4,<2.0.0
20
+ # scipy>=1.15.0 # Removed - system package conflict
21
 
22
  # Data handling
23
  dm-tree>=0.1.8
 
41
  treescope>=0.1.7
42
 
43
  # File handling and storage
44
+ # fsspec[gcs]>=2024.6.0 # Removed - system package conflict
45
  flatbuffers>=24.3.25
46
  h5py>=3.10.0
47
 
 
58
  websockets>=11.0
59
 
60
  # Development (minimal - remove if not needed)
61
+ # typing-extensions>=4.12.2 # Removed - system package conflict
62
+ pytest>=6.0.0
63
 
64
  # Robotics/Control specific
65
  gymnasium>=0.29.0
66
  mujoco>=2.3.0
67
+