diff --git a/pyproject.toml b/pyproject.toml index 562e9ba27..efaaa8563 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -34,7 +34,7 @@ dependencies = [ "draccus==0.8.0", "einops", # "flash_attn==2.5.5", # Here for documentation -- install *AFTER* editable install (follow README) - "huggingface_hub", + "huggingface_hub==0.21.0", "json-numpy", "jsonlines", "matplotlib", @@ -53,7 +53,7 @@ dependencies = [ "tensorflow_datasets==4.9.3", "tensorflow_graphics==2021.12.3", "dlimp @ git+https://github.com/moojink/dlimp_openvla", - "diffusers", + "diffusers==0.27.2", "imageio", "uvicorn", "fastapi",