jokester commited on
Commit
2d49fdd
·
0 Parent(s):

start over with minimal settings

Browse files
Files changed (3) hide show
  1. Dockerfile +11 -0
  2. conda.yaml +12 -0
  3. requirements-moeflow.txt +54 -0
Dockerfile ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM mambaorg/micromamba:2-debian12-slim
2
+
3
+ COPY ./conda.yaml /tmp/conda.yaml
4
+ RUN --mount=type=cache,uid=57439,gid=57439,target=/opt/conda/pkgs micromamba env create --yes --file /tmp/conda.yaml
5
+ # RUN apt update \
6
+ # && DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get -y install tzdata build-essential
7
+ COPY --chown=mambauser . /app
8
+ WORKDIR /app
9
+ RUN micromamba run -n mit-py311 python3 -mvenv --system-site-packages venv
10
+ RUN venv/bin/pip install -r requirements-moeflow.txt
11
+ RUN venv/bin/python docker_prepare.py --models ocr.48px,ocr.48px_ctc,ocr.32px,ocr.mocr
conda.yaml ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: mit-py311
2
+ channels:
3
+ - conda-forge
4
+ - pytorch
5
+ - nvidia
6
+ dependencies:
7
+ - python==3.11
8
+ - pytorch==2.2.2
9
+ - torchvision==0.17.2
10
+ - torchaudio==2.2.2
11
+ - pytorch-cuda=12.1
12
+ - numpy<2
requirements-moeflow.txt ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # vim: ft=requirements
2
+ # networkx
3
+ # torch
4
+ # groq
5
+ # torchvision
6
+ # torch-summary
7
+ einops
8
+ scikit-image
9
+ opencv-python
10
+ pyclipper
11
+ shapely
12
+ # requests
13
+ # cryptography
14
+ # freetype-py
15
+ # aiohttp
16
+ tqdm
17
+ # deepl
18
+ # ImageHash
19
+ # kornia
20
+ # backports.cached-property
21
+ # huggingface_hub
22
+ # transformers
23
+ py3langid==0.2.2
24
+ # sentencepiece
25
+ # editdistance
26
+ # numpy==1.26.4
27
+ # tensorboardX
28
+ # websockets
29
+ # protobuf
30
+ # ctranslate2
31
+ colorama
32
+ # openai==1.35.9
33
+ # open_clip_torch
34
+ # safetensors
35
+ pandas
36
+ onnxruntime==1.18.1
37
+ timm
38
+ omegaconf
39
+ python-dotenv
40
+ # nest-asyncio
41
+ # marshmallow
42
+ # cython
43
+ # aioshutil
44
+ # aiofiles
45
+ # arabic-reshaper
46
+ # pyhyphen
47
+ # langcodes
48
+ manga-ocr
49
+ # langdetect
50
+ # pydensecrf@https://github.com/lucasb-eyer/pydensecrf/archive/refs/heads/master.zip
51
+ # accelerate
52
+ # bitsandbytes
53
+ gradio==5.6.0
54
+ orjson==3.10.11