Add pusht on hf dataset (WIP)

2024-03-14 16:59:37 +00:00
parent 95e84079ef
commit 19730b3412
3 changed files with 7 additions and 1 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -50,6 +50,7 @@ diffusers = "^0.26.3"
 torchvision = "^0.17.1"
 h5py = "^3.10.0"
 dm-control = "1.0.14"
+huggingface-hub = "^0.21.4"


 [tool.poetry.group.dev.dependencies]