XiangpengYang commited on
Commit
6408eec
·
1 Parent(s): 45e8b30

fix: use ZeroGPU-supported torch version

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -1
  2. 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.7.1
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()