nexusbert commited on
Commit
08006d6
·
1 Parent(s): d60836c

fix: Update MINIMAX_LOW_VRAM to True for improved performance

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. 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 = False
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