Skip to content

Commit

Permalink
common install steps for mac and linux
Browse files Browse the repository at this point in the history
Signed-off-by: James Busche <[email protected]>
  • Loading branch information
jbusche committed Mar 9, 2024
1 parent 0aadf3d commit bf90df9
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion CONTRIBUTING.md
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ The following tools are required:
Installation:
```
pip install -U datasets
pip install -e .
pip install -e "."
```

### Unit tests
Expand Down
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@ pip install -e .

> Note: After installing, if you wish to use [FlashAttention](https://github.com/Dao-AILab/flash-attention), then you need to install these requirements:
'''
pip install -e ."[dev]"
pip install -e ."[flash-attn]
pip install -e ".[dev]"
pip install -e ".[flash-attn]"
```
[FlashAttention](https://github.com/Dao-AILab/flash-attention) requires the [CUDA Toolit](https://developer.nvidia.com/cuda-toolkit) to be pre-installed.
Expand Down
4 changes: 2 additions & 2 deletions build/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -101,8 +101,8 @@ RUN python -m pip install --upgrade pip
# TODO Move to installing wheel once we have proper releases setup instead of cloning the repo
RUN git clone https://github.com/foundation-model-stack/fms-hf-tuning.git && \
cd fms-hf-tuning && \
python -m pip install .[dev] && \
python -m pip install .[flash-attn] && \
python -m pip install ".[dev]" && \
python -m pip install ".[flash-attn]" && \
python -m pip install -U datasets

RUN mkdir -p /licenses
Expand Down

0 comments on commit bf90df9

Please sign in to comment.