Enable qwen-image (#56)

Author: hzjane

visual-ai/Qwen-image/README.md

@@ -0,0 +1,39 @@
+# Docker setup
+
+Build docker image:
+
+```bash
+bash build.sh
+```
+
+Run docker image:
+
+```bash
+export DOCKER_IMAGE=llm-scaler-visualai:qwen-image
+export CONTAINER_NAME=qwen-image
+export MODEL_DIR=<your_model_dir>
+sudo docker run -itd \
+        --privileged \
+        --net=host \
+        --device=/dev/dri \
+        -e no_proxy=localhost,127.0.0.1 \
+        --name=$CONTAINER_NAME \
+        -v $MODEL_DIR:/llm/models/ \
+        --shm-size="16g" \
+        --entrypoint=/bin/bash \
+        $DOCKER_IMAGE
+
+docker exec -it qwen-image /bin/bash
+```
+
+Download models before start example.
+Run qwen-image demo on Single GPU:
+```bash
+python3 qwen_image_example.py
+```
+
+Run qwen-image demo on two GPUs:
+```bash
+export qwen_image_enable_two_card=1
+python3 qwen_image_example.py
+```

visual-ai/Qwen-image/build.sh

@@ -0,0 +1,4 @@
+export HTTP_PROXY=<your_http_proxy>
+export HTTPS_PROXY=<your_https_proxy>
+
+docker build -f ./docker/Dockerfile . -t llm-scaler-visualai:qwen-image --build-arg https_proxy=$HTTPS_PROXY --build-arg http_proxy=$HTTP_PROXY

visual-ai/Qwen-image/docker/Dockerfile

@@ -0,0 +1,70 @@
+# Copyright (C) 2025 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
+# ======== Base Stage ========
+FROM intel/deep-learning-essentials:2025.0.2-0-devel-ubuntu24.04 AS vllm-base
+
+ARG https_proxy
+ARG http_proxy
+
+# Add Intel oneAPI repo and PPA for GPU support
+RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | gpg --dearmor | tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null && \
+    echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" | tee /etc/apt/sources.list.d/oneAPI.list && \
+    add-apt-repository -y ppa:kobuk-team/intel-graphics-testing
+
+# Install dependencies and Python 3.10
+RUN apt-get update -y && \
+    apt-get install -y software-properties-common && \
+    add-apt-repository ppa:deadsnakes/ppa && \
+    apt-get update -y && \
+    apt-get install -y python3.10 python3.10-distutils python3.10-dev && \
+    curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10 && \
+    apt-get install -y --no-install-recommends --fix-missing \
+        curl \
+        ffmpeg \
+        git \
+        libsndfile1 \
+        libsm6 \
+        libxext6 \
+        libgl1 \
+        lsb-release \
+        numactl \
+        wget \
+        vim \
+        linux-libc-dev && \
+    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 1 && \
+    # Install Intel GPU runtime packages
+    apt-get update -y && \
+    apt-get install -y libze1 libze-dev libze-intel-gpu1 intel-opencl-icd libze-intel-gpu-raytracing && \
+    apt-get install -y intel-oneapi-dpcpp-ct=2025.0.1-17 && \
+    apt-get clean && rm -rf /var/lib/apt/lists/*
+
+WORKDIR /llm
+
+# Set environment variables early
+ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/lib/"
+
+# ======= Add oneCCL build =======
+RUN apt-get update && apt-get install -y \
+    cmake \
+    g++ \
+    && rm -rf /var/lib/apt/lists/*
+
+ENV LD_LIBRARY_PATH="/usr/local/lib:/usr/local/lib/python3.10/dist-packages/torch/lib:$LD_LIBRARY_PATH"
+
+RUN pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/xpu && \
+    pip install tqdm huggingface_hub modelscope einops transformers==4.53.3 sentencepiece && \
+    apt remove python3-blinker -y 
+
+COPY ./patches/qwen_image_for_multi_arc.patch /tmp/
+COPY ./qwen_image_example.py /llm/qwen-image/
+
+RUN cd /llm && \
+    git clone https://github.com/modelscope/DiffSynth-Studio.git && \
+    cd ./DiffSynth-Studio && \
+    git checkout 89bf3ce5cfe1ed5a9aa3bd2c2f74da0a82ba84f4 && \
+    git apply /tmp/qwen_image_for_multi_arc.patch && \
+    pip install -e .
+
+
+WORKDIR /llm/qwen-image/