diff --git a/examples/pi0/README.md b/examples/pi0/README.md index 13f02422d..44d7819c0 100644 --- a/examples/pi0/README.md +++ b/examples/pi0/README.md @@ -25,7 +25,8 @@ Install FlagScale and robotics dependencies: ```sh cd FlagScale/ pip install . --verbose -pip install -r requirements/train/robotics/requirements.txt +pip install -r requirements/cuda/base.txt +pip install git+https://github.com/huggingface/transformers.git@fix/lerobot_openpi ``` Install additional dependencies for downloading models/datasets: diff --git a/examples/pi0_5/README.md b/examples/pi0_5/README.md index e345d25e4..5758999e0 100644 --- a/examples/pi0_5/README.md +++ b/examples/pi0_5/README.md @@ -25,7 +25,8 @@ Install FlagScale and robotics dependencies: ```sh cd FlagScale/ pip install . --verbose -pip install -r requirements/train/robotics/requirements.txt +pip install -r requirements/cuda/base.txt +pip install git+https://github.com/huggingface/transformers.git@fix/lerobot_openpi ``` Install additional dependencies for downloading models/datasets: diff --git a/examples/robobrain2_5/README.md b/examples/robobrain2_5/README.md index 05e1275f3..2c76b1562 100644 --- a/examples/robobrain2_5/README.md +++ b/examples/robobrain2_5/README.md @@ -19,20 +19,21 @@ conda create -n flagscale-inference python=3.12 conda activate flagscale-inference ``` -Install FlagScale: +### Install FlagScale ```sh cd FlagScale/ pip install . --verbose +pip install -r requirements/common.txt ``` -### Install vLLM and Transformers +### Install vllm-plugin-FL + +Follow Setup in the vllm-plugin-FL [setup guide](https://github.com/flagos-ai/vllm-plugin-FL?tab=readme-ov-file#setup). + +### Install Transformers ```sh -git clone https://github.com/flagos-ai/vllm-FL.git -cd vllm-FL -pip install packaging==24.2 -pip install --no-build-isolation . pip install transformers==4.57.0 ``` diff --git a/requirements/cuda/base.txt b/requirements/cuda/base.txt index 7bd2ea205..e6a8e5e4d 100644 --- a/requirements/cuda/base.txt +++ b/requirements/cuda/base.txt @@ -7,3 +7,14 @@ torch==2.9.1 torchaudio==2.9.1 torchvision==0.24.1 + +# for robotics +webdataset==1.0.2 +epath==0.7 +qwen_vl_utils==0.0.14 +diffusers==0.36.0 +flask_cors==6.0.2 +datasets==4.5.0 +draccus==0.11.5 +einops==0.8.2 +multi-storage-client==0.42.0