commit 6e6aa3b2754d78194678ecdc6079917a9bf96134 Author: jianjiang Date: Fri Apr 21 16:05:43 2023 +0800 ok diff --git a/.gitea/workflows/build.yaml b/.gitea/workflows/build.yaml new file mode 100644 index 0000000..03be22b --- /dev/null +++ b/.gitea/workflows/build.yaml @@ -0,0 +1,47 @@ +name: Build +run-name: ${{ github.actor }} is upgrade release 🚀 +on: [push] +env: + REPOSITORY: ${{ github.repository }} + COMMIT_ID: ${{ github.sha }} +jobs: + Build-Deploy-Actions: + runs-on: ubuntu-latest + steps: + - run: echo "🎉 The job was automatically triggered by a ${{ github.event_name }} event." + - run: echo "🐧 This job is now running on a ${{ runner.os }} server hosted by Gitea!" + - run: echo "🔎 The name of your branch is ${{ github.ref }} and your repository is ${{ github.repository }}." + - name: Check out repository code + uses: actions/checkout@v3 + - + name: Setup Git LFS + run: | + git lfs install + git lfs fetch + git lfs checkout + - name: List files in the repository + run: | + ls ${{ github.workspace }} + - + name: Docker Image Info + id: image-info + run: | + echo "::set-output name=image_name::$(echo $REPOSITORY | tr '[:upper:]' '[:lower:]')" + echo "::set-output name=image_tag::${COMMIT_ID:0:10}" + - + name: Login to Docker Hub + uses: docker/login-action@v2 + with: + registry: artifacts.iflytek.com + username: ${{ secrets.DOCKERHUB_USERNAME }} + password: ${{ secrets.DOCKERHUB_TOKEN }} + - name: Set up Docker Buildx + uses: docker/setup-buildx-action@v2 + - + name: Build and push + run: | + docker version + docker buildx build -t artifacts.iflytek.com/docker-private/atp/${{ steps.image-info.outputs.image_name }}:${{ steps.image-info.outputs.image_tag }} . --file ${{ github.workspace }}/Dockerfile --load + docker push artifacts.iflytek.com/docker-private/atp/${{ steps.image-info.outputs.image_name }}:${{ steps.image-info.outputs.image_tag }} + docker rmi artifacts.iflytek.com/docker-private/atp/${{ steps.image-info.outputs.image_name }}:${{ steps.image-info.outputs.image_tag }} + - run: echo "🍏 This job's status is ${{ job.status }}." diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..d82fe3e --- /dev/null +++ b/Dockerfile @@ -0,0 +1,24 @@ +# please visit https://github.com/xfyun/aiges/releases to get stable and suitable iamges. + +FROM iflyopensource/aiges-gpu:11.2-1.17-3.9.13-ubuntu1804-v3.3.2 + +RUN sed -i 's@//.*archive.ubuntu.com@//mirrors.ustc.edu.cn@g' /etc/apt/sources.list +RUN sed -i 's/security.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list + +RUN --mount=target=/root/packages.txt,source=packages.txt apt-get update && xargs -r -a /root/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/* + +WORKDIR /home/user/app +RUN useradd -m -u 1000 user +RUN chown -R 1000.1000 /home/user + + +RUN pip config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/ + + +RUN pip install --no-cache-dir pip==22.3.1 +RUN --mount=target=requirements.txt,source=requirements.txt pip install --no-cache-dir -r requirements.txt + +COPY --chown=1000 ./ /home/user/app + + +CMD ["python3", "app.py"] diff --git a/app.py b/app.py new file mode 100644 index 0000000..278196b --- /dev/null +++ b/app.py @@ -0,0 +1,23 @@ +from diffusers.models import AutoencoderKL +from diffusers import StableDiffusionPipeline +import gradio as gr + +model = "CompVis/stable-diffusion-v1-4" +vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse") +pipe = StableDiffusionPipeline.from_pretrained(model, vae=vae) + +def text2image(prompt): + image = pipe(prompt).images[0] + + return image + + +demo = gr.Interface(fn=text2image, + inputs='text', + outputs='image', + title = "text2image", + examples = ['a photo of an astronaut riding a horse on mars']) + + +if __name__ == "__main__": + demo.queue(concurrency_count=3).launch(server_name = "0.0.0.0") diff --git a/packages.txt b/packages.txt new file mode 100644 index 0000000..b399c93 --- /dev/null +++ b/packages.txt @@ -0,0 +1,2 @@ +git +curl \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..b653526 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,4 @@ +gradio==3.21.0 +diffusers +transformers==4.27.1 +torch==2.0.0