alishafique commited on
Commit
8315568
·
verified ·
1 Parent(s): 409de86

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -2
Dockerfile CHANGED
@@ -22,8 +22,7 @@ RUN nvcc --version && python3 --version
22
  # Install depencencies
23
  RUN python3 -m pip install --upgrade pip pytest cmake \
24
  scikit-build setuptools fastapi uvicorn sse-starlette \
25
- pydantic-settings starlette-context gradio huggingface_hub hf_transfer \
26
- transformers==4.38.2 accelerate bitsandbytes optimum
27
 
28
  # Install llama-cpp-python (build with cuda)
29
  # RUN CMAKE_ARGS="-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=75" FORCE_CMAKE=1 python3 -m pip install llama-cpp-python --force-reinstall --upgrade --no-cache-dir --verbose
 
22
  # Install depencencies
23
  RUN python3 -m pip install --upgrade pip pytest cmake \
24
  scikit-build setuptools fastapi uvicorn sse-starlette \
25
+ pydantic-settings starlette-context gradio huggingface_hub hf_transfer
 
26
 
27
  # Install llama-cpp-python (build with cuda)
28
  # RUN CMAKE_ARGS="-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=75" FORCE_CMAKE=1 python3 -m pip install llama-cpp-python --force-reinstall --upgrade --no-cache-dir --verbose