matitie commited on
Commit
7d6b687
·
verified ·
1 Parent(s): 6807ca7

Build public PickStyle worker runtime image

Browse files
Files changed (3) hide show
  1. Dockerfile +18 -0
  2. README.md +9 -4
  3. requirements.txt +30 -0
Dockerfile ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04
2
+
3
+ ENV DEBIAN_FRONTEND=noninteractive \
4
+ PATH=/opt/venv/bin:$PATH \
5
+ PYTHONUNBUFFERED=1
6
+
7
+ RUN apt-get update \
8
+ && apt-get install -y --no-install-recommends ffmpeg python3 python3-venv ca-certificates \
9
+ && rm -rf /var/lib/apt/lists/* \
10
+ && python3 -m venv /opt/venv
11
+
12
+ WORKDIR /runtime
13
+ COPY requirements.txt /runtime/requirements.txt
14
+ RUN python3 -m pip install --no-cache-dir --upgrade pip \
15
+ && python3 -m pip install --no-cache-dir -r /runtime/requirements.txt
16
+
17
+ EXPOSE 7860
18
+ CMD python3 -m http.server 7860 --bind 0.0.0.0
README.md CHANGED
@@ -1,10 +1,15 @@
1
  ---
2
  title: PickStyle Worker Image
3
- emoji: 🏆
4
- colorFrom: pink
5
- colorTo: red
6
  sdk: docker
 
7
  pinned: false
8
  ---
9
 
10
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
1
  ---
2
  title: PickStyle Worker Image
3
+ emoji: 🧰
4
+ colorFrom: blue
5
+ colorTo: gray
6
  sdk: docker
7
+ app_port: 7860
8
  pinned: false
9
  ---
10
 
11
+ # PickStyle Worker Image
12
+
13
+ Public, dependency-only runtime image for ephemeral GPU Jobs. The container exposes
14
+ a minimal health endpoint; application source and data are mounted separately at
15
+ job startup.
requirements.txt ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ torch==2.8.0
2
+ torchvision==0.23.0
3
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3%2Bcu12torch2.8cxx11abiFALSE-cp312-cp312-linux_x86_64.whl
4
+ gradio==6.20.0
5
+ huggingface_hub>=1.8,<2
6
+ diffusers>=0.31.0
7
+ transformers>=4.49.0,<6
8
+ tokenizers>=0.20.3
9
+ accelerate>=1.1.1
10
+ numpy>=1.23.5,<2
11
+ opencv-python-headless>=4.9.0.80
12
+ decord
13
+ imageio
14
+ imageio-ffmpeg
15
+ av
16
+ einops
17
+ easydict
18
+ ftfy
19
+ packaging
20
+ peft
21
+ Pillow
22
+ regex
23
+ safetensors
24
+ scipy
25
+ sentencepiece
26
+ tqdm
27
+ fastapi>=0.115,<1
28
+ httpx>=0.27,<1
29
+ pydantic>=2.10,<3
30
+ uvicorn>=0.34,<1