Hugging Face's logo

Portx
/
trial-multi-model

Model card Files
xet
 Community
Portx commited on
Commit
72363cc
·
verified ·
1 Parent(s): b4facea

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -0
handler.py CHANGED
@@ -6,6 +6,8 @@ import os
6
 
7
  #run("pip install flash-attn --no-build-isolation", shell=True, check=True)
8
  run("pip install --upgrade pip", shell=True, check=True)
 
 
9
  #run("pip install --upgrade accelerate transformers", shell=True, check=True)
10
  #run("pip -qqq install --force-reinstall https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_main/bitsandbytes-0.44.2.dev0-py3-none-manylinux_2_24_x86_64.whl", shell=True, check=True)
11
 
 
6
 
7
  #run("pip install flash-attn --no-build-isolation", shell=True, check=True)
8
  run("pip install --upgrade pip", shell=True, check=True)
9
+ run("pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu124", shell=True, check=True)
10
+
11
  #run("pip install --upgrade accelerate transformers", shell=True, check=True)
12
  #run("pip -qqq install --force-reinstall https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_main/bitsandbytes-0.44.2.dev0-py3-none-manylinux_2_24_x86_64.whl", shell=True, check=True)
13