gokaygokay commited on
Commit
75214fd
·
verified ·
1 Parent(s): c52b1a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def install_packages():
31
 
32
  # Set environment variables for CUDA support
33
  os.environ["CUDA_HOME"] = "/usr/local/cuda" # Adjust this path if needed
34
- os.environ["CMAKE_ARGS"] = "-DLLAMA_CUBLAS=on"
35
  os.environ["FORCE_CMAKE"] = "1"
36
 
37
  # Install all optional dependencies with CUDA support
 
31
 
32
  # Set environment variables for CUDA support
33
  os.environ["CUDA_HOME"] = "/usr/local/cuda" # Adjust this path if needed
34
+ os.environ["CMAKE_ARGS"] = "-GGML_CUDA=on"
35
  os.environ["FORCE_CMAKE"] = "1"
36
 
37
  # Install all optional dependencies with CUDA support