support diffusers ms models

2025-09-08 16:32:50 +08:00
parent 0ae0bd7904
commit 495c3fcd8a
6 changed files with 47 additions and 7 deletions
--- a/Dockerfile.bi100
+++ b/Dockerfile.bi100
@@ -0,0 +1,7 @@
+FROM git.modelhub.org.cn:980/enginex-iluvatar/bi100-3.2.1-x86-ubuntu20.04-py3.10-poc-llm-infer:v1.2.2
+
+WORKDIR /workspace
+
+RUN pip install diffusers==0.34.0
+RUN pip install imageio[ffmpeg] einops datasets==3.2.0 simplejson addict open_clip_torch==2.24.0 sortedcontainers modelscope==1.28.2 av==11.0.0 pytorch-lightning
+COPY . /workspace/