Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|