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
|