|
|
@@ -145,7 +145,8 @@ ENV BACKEND=${BACKEND} \
|
|
|
PYTHONUNBUFFERED=1 \
|
|
|
UV_HTTP_TIMEOUT=3600 \
|
|
|
UV_INDEX_URL=http://mirrors.aliyun.com/pypi/simple/ \
|
|
|
- UV_DEFAULT_INDEX=http://mirrors.aliyun.com/pypi/simple/
|
|
|
+ UV_DEFAULT_INDEX=http://mirrors.aliyun.com/pypi/simple/ \
|
|
|
+ PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
|
|
|
|
|
|
# System dependencies for audio processing
|
|
|
ARG DEPENDENCIES=" \
|
