添加对accelerate库的依赖的说明

This commit is contained in:
HfCloud 2023-02-06 02:06:01 +08:00
parent c4acedc22f
commit 3c7cc5ab0c
1 changed files with 1 additions and 1 deletions

View File

@ -24,7 +24,7 @@
python3需要再安装这些库使用pip安装就可以
```
pip install torch flask openai transformers diffusers python-opencv
pip install torch flask openai transformers diffusers python-opencv accelerate
```
当然如果使用cuda加速建议按照<a href="https://pytorch.org">pytorch官网</a>提供的方法安装支持cuda加速的torch版本。