Skip to content

Commit

Permalink
update requirements (#280)
Browse files Browse the repository at this point in the history
  • Loading branch information
zhengzangw authored Apr 15, 2024
1 parent a5afed2 commit 436ee2c
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -123,10 +123,12 @@ conda activate opensora
pip install torch torchvision

# install flash attention (optional)
# set enable_flashattn=False in config to avoid using flash attention
pip install packaging ninja
pip install flash-attn --no-build-isolation

# install apex (optional)
# set enable_layernorm_kernel=False in config to avoid using apex
pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" git+https://github.com/NVIDIA/apex.git

# install xformers
Expand Down

0 comments on commit 436ee2c

Please sign in to comment.