Spaces:
Sleeping
Sleeping
Commit ·
6408eec
1
Parent(s): 45e8b30
fix: use ZeroGPU-supported torch version
Browse files- requirements.txt +1 -1
- tests/test_space_config.py +2 -0
requirements.txt
CHANGED
|
@@ -16,7 +16,7 @@ numpydantic>=1.6.6
|
|
| 16 |
opencv-python-headless>=4.10.0.84
|
| 17 |
orbax-checkpoint==0.11.13
|
| 18 |
sentencepiece>=0.2.0
|
| 19 |
-
torch==2.
|
| 20 |
typing-extensions>=4.12.2
|
| 21 |
tyro>=0.9.5
|
| 22 |
wandb>=0.19.1
|
|
|
|
| 16 |
opencv-python-headless>=4.10.0.84
|
| 17 |
orbax-checkpoint==0.11.13
|
| 18 |
sentencepiece>=0.2.0
|
| 19 |
+
torch==2.11.0
|
| 20 |
typing-extensions>=4.12.2
|
| 21 |
tyro>=0.9.5
|
| 22 |
wandb>=0.19.1
|
tests/test_space_config.py
CHANGED
|
@@ -11,6 +11,8 @@ def test_space_metadata_and_runtime_files():
|
|
| 11 |
assert "python_version: '3.11'" in readme
|
| 12 |
assert "app_file: app.py" in readme
|
| 13 |
assert "gradio==6.20.0" in requirements
|
|
|
|
|
|
|
| 14 |
assert "lerobot.git@0cf864870cf29f4738d3ade893e6fd13fbd7cdb5" in requirements
|
| 15 |
assert (ROOT / "openpi_runtime/openpi/policies/ur_policy.py").is_file()
|
| 16 |
assert (ROOT / "openpi_runtime/openpi_client/base_policy.py").is_file()
|
|
|
|
| 11 |
assert "python_version: '3.11'" in readme
|
| 12 |
assert "app_file: app.py" in readme
|
| 13 |
assert "gradio==6.20.0" in requirements
|
| 14 |
+
assert "torch==2.11.0" in requirements
|
| 15 |
+
assert "torch==2.7.1" not in requirements
|
| 16 |
assert "lerobot.git@0cf864870cf29f4738d3ade893e6fd13fbd7cdb5" in requirements
|
| 17 |
assert (ROOT / "openpi_runtime/openpi/policies/ur_policy.py").is_file()
|
| 18 |
assert (ROOT / "openpi_runtime/openpi_client/base_policy.py").is_file()
|