Update requirements.txt
Browse files- 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 |
+
|