fix: Update MINIMAX_LOW_VRAM to True for improved performance
Browse files- Dockerfile +2 -0
- app/core/config.py +1 -1
Dockerfile
CHANGED
|
@@ -3,6 +3,8 @@ FROM python:3.11-slim
|
|
| 3 |
|
| 4 |
WORKDIR /app
|
| 5 |
|
|
|
|
|
|
|
| 6 |
# Install system dependencies
|
| 7 |
RUN apt-get update && apt-get install -y \
|
| 8 |
build-essential \
|
|
|
|
| 3 |
|
| 4 |
WORKDIR /app
|
| 5 |
|
| 6 |
+
ENV PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
|
| 7 |
+
|
| 8 |
# Install system dependencies
|
| 9 |
RUN apt-get update && apt-get install -y \
|
| 10 |
build-essential \
|
app/core/config.py
CHANGED
|
@@ -19,7 +19,7 @@ class Settings(BaseSettings):
|
|
| 19 |
MINIMAX_SERVER_URL: Optional[str] = None
|
| 20 |
MINIMAX_AUDIO_DURATION_SECONDS: float = 60.0
|
| 21 |
MINIMAX_SEED: int = 7
|
| 22 |
-
MINIMAX_LOW_VRAM: bool =
|
| 23 |
|
| 24 |
CORS_ORIGINS: List[str] = ["*"]
|
| 25 |
|
|
|
|
| 19 |
MINIMAX_SERVER_URL: Optional[str] = None
|
| 20 |
MINIMAX_AUDIO_DURATION_SECONDS: float = 60.0
|
| 21 |
MINIMAX_SEED: int = 7
|
| 22 |
+
MINIMAX_LOW_VRAM: bool = True
|
| 23 |
|
| 24 |
CORS_ORIGINS: List[str] = ["*"]
|
| 25 |
|