Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -23,6 +23,7 @@ RUN apt-get update && apt-get install -y g++ build-essential
|
|
| 23 |
|
| 24 |
|
| 25 |
ENV TORCHINDUCTOR_CACHE_DIR="/tmp/torch_cache"
|
|
|
|
| 26 |
|
| 27 |
# Create and set permissions for cache directories
|
| 28 |
RUN mkdir -p /.cache && chmod -R 777 /.cache
|
|
|
|
| 23 |
|
| 24 |
|
| 25 |
ENV TORCHINDUCTOR_CACHE_DIR="/tmp/torch_cache"
|
| 26 |
+
ENV BNB_CUDA_OFFLOAD=0
|
| 27 |
|
| 28 |
# Create and set permissions for cache directories
|
| 29 |
RUN mkdir -p /.cache && chmod -R 777 /.cache
|