Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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"] = "-
|
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
|