Forráskód Böngészése

install hivemind from pip

justheuristic 3 éve
szülő
commit
14b6d04b0f
1 módosított fájl, 1 hozzáadás és 2 törlés
  1. 1 2
      README.md

+ 1 - 2
README.md

@@ -17,9 +17,8 @@ conda activate bloom-demo
 
 conda install -y -c conda-forge cudatoolkit-dev==11.3.1 cudatoolkit==11.3.1 cudnn==8.2.1.32
 pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
-pip install accelerate==0.10.0 huggingface-hub==0.7.0
+pip install accelerate==0.10.0 huggingface-hub==0.7.0 hivemind==1.1.0
 pip install bitsandbytes-cuda113==0.26.0
-pip install https://github.com/learning-at-home/hivemind/archive/master.zip
 pip install https://github.com/huggingface/transformers/archive/6589e510fa4e6c442059de2fab84752535de9b23.zip
 ```