Amirhustler commited on
Commit
d0387d9
·
verified ·
1 Parent(s): 7c8c7ca

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +13 -31
Dockerfile CHANGED
@@ -1,39 +1,21 @@
1
- FROM python:3.10-slim
2
-
3
- ENV PYTHONUNBUFFERED=1 \
4
- PIP_PROGRESS_BAR=on \
5
- PIP_DEFAULT_TIMEOUT=120 \
6
- PIP_NO_CACHE_DIR=1 \
7
- HF_HOME=/app/.cache/huggingface \
8
- XDG_CACHE_HOME=/app/.cache \
9
- MPLCONFIGDIR=/app/.config/matplotlib \
10
- OPENVINO_TELEMETRY=NO \
11
- OV_TELEMETRY=0 \
12
- HF_HUB_ENABLE_TELEMETRY=0 \
13
- HOME=/app
14
-
15
- RUN apt-get update && apt-get install -y --no-install-recommends \
16
- git ca-certificates curl && \
17
- rm -rf /var/lib/apt/lists/*
18
-
19
- RUN mkdir -p /app /app/.cache/huggingface /app/.cache /app/.config/matplotlib
20
- WORKDIR /app
21
-
22
- COPY requirements.txt ./requirements.txt
23
- RUN python -V && pip -V
24
  RUN pip install -r requirements.txt -v && \
25
  python - <<'PY'
26
  print(">>> sanity import …")
27
- import openvino, transformers, diffusers, optimum
28
- from optimum.intel.openvino import OVStableDiffusionPipeline
 
 
 
29
  print("OpenVINO:", openvino.__version__)
30
  print("Transformers:", transformers.__version__)
31
  print("Diffusers:", diffusers.__version__)
32
- print("Optimum:", optimum.__version__)
 
 
 
 
 
 
 
33
  print("OK: OVStableDiffusionPipeline import passed")
34
  PY
35
-
36
- COPY app.py ./app.py
37
-
38
- EXPOSE 7860
39
- CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  RUN pip install -r requirements.txt -v && \
2
  python - <<'PY'
3
  print(">>> sanity import …")
4
+ import importlib
5
+ import openvino, transformers, diffusers
6
+ from importlib.metadata import version, PackageNotFoundError
7
+
8
+ # چاپ نسخه‌ها
9
  print("OpenVINO:", openvino.__version__)
10
  print("Transformers:", transformers.__version__)
11
  print("Diffusers:", diffusers.__version__)
12
+ try:
13
+ print("Optimum:", version("optimum"))
14
+ except PackageNotFoundError:
15
+ print("Optimum: <not found>")
16
+
17
+ # تست مهم: آیا کلاس OpenVINO لود می‌شود؟
18
+ m = importlib.import_module("optimum.intel.openvino")
19
+ from optimum.intel.openvino import OVStableDiffusionPipeline
20
  print("OK: OVStableDiffusionPipeline import passed")
21
  PY