gokaygokay commited on
Commit
490cf43
·
verified ·
1 Parent(s): 8774fad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,11 +31,11 @@ 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"] = "-DGGML_CUDA=on"
35
  os.environ["FORCE_CMAKE"] = "1"
36
 
37
  # Install all optional dependencies with CUDA support
38
- run_command("pip install -e .[all]")
39
 
40
  run_command("make clean")
41
 
 
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"] = "-DGGML_CUDA=on -DCUDAToolkit_ROOT=/usr/local/cuda" # Add CUDAToolkit_ROOT
35
  os.environ["FORCE_CMAKE"] = "1"
36
 
37
  # Install all optional dependencies with CUDA support
38
+ run_command("pip install -e .[all] --verbose") # Add --verbose for more detailed output
39
 
40
  run_command("make clean")
41