Felladrin commited on
Commit
77cbaf0
·
1 Parent(s): 5fff74f

Remove unnecessary environment variables

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -3
  2. app.py +1 -4
Dockerfile CHANGED
@@ -1,8 +1,6 @@
1
  FROM python:3.12-slim
2
 
3
- ENV PYTHONDONTWRITEBYTECODE=1 \
4
- PYTHONUNBUFFERED=1 \
5
- PORT=8501
6
 
7
  RUN apt-get update && apt-get install -y --no-install-recommends \
8
  git git-lfs ffmpeg libsm6 libxext6 cmake libgl1 build-essential curl \
 
1
  FROM python:3.12-slim
2
 
3
+ ENV PORT=8501
 
 
4
 
5
  RUN apt-get update && apt-get install -y --no-install-recommends \
6
  git git-lfs ffmpeg libsm6 libxext6 cmake libgl1 build-essential curl \
app.py CHANGED
@@ -285,7 +285,7 @@ class ModelConverter:
285
  sys.executable,
286
  "-m",
287
  "scripts.convert",
288
- "--quantize", # Enable quantization for smaller model size
289
  "--model_id",
290
  input_model_id,
291
  ]
@@ -301,9 +301,6 @@ class ModelConverter:
301
  text=True,
302
  env={
303
  "HF_TOKEN": self.config.hf_token,
304
- # Force eager attention implementation for compatibility
305
- "TRANSFORMERS_ATTENTION_IMPLEMENTATION": "eager",
306
- "PYTORCH_SDP_KERNEL": "math",
307
  },
308
  )
309
 
 
285
  sys.executable,
286
  "-m",
287
  "scripts.convert",
288
+ "--quantize",
289
  "--model_id",
290
  input_model_id,
291
  ]
 
301
  text=True,
302
  env={
303
  "HF_TOKEN": self.config.hf_token,
 
 
 
304
  },
305
  )
306