Spaces:
Running
on
Zero
Running
on
Zero
Update
Browse files- Dockerfile +0 -53
- README.md +2 -1
- app.py +2 -0
- requirements.txt +3 -5
Dockerfile
DELETED
@@ -1,53 +0,0 @@
|
|
1 |
-
FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
|
2 |
-
ENV DEBIAN_FRONTEND=noninteractive
|
3 |
-
RUN apt-get update && \
|
4 |
-
apt-get upgrade -y && \
|
5 |
-
apt-get install -y --no-install-recommends \
|
6 |
-
git \
|
7 |
-
git-lfs \
|
8 |
-
wget \
|
9 |
-
curl \
|
10 |
-
# python build dependencies \
|
11 |
-
build-essential \
|
12 |
-
libssl-dev \
|
13 |
-
zlib1g-dev \
|
14 |
-
libbz2-dev \
|
15 |
-
libreadline-dev \
|
16 |
-
libsqlite3-dev \
|
17 |
-
libncursesw5-dev \
|
18 |
-
xz-utils \
|
19 |
-
tk-dev \
|
20 |
-
libxml2-dev \
|
21 |
-
libxmlsec1-dev \
|
22 |
-
libffi-dev \
|
23 |
-
liblzma-dev && \
|
24 |
-
apt-get clean && \
|
25 |
-
rm -rf /var/lib/apt/lists/*
|
26 |
-
|
27 |
-
RUN useradd -m -u 1000 user
|
28 |
-
USER user
|
29 |
-
ENV HOME=/home/user \
|
30 |
-
PATH=/home/user/.local/bin:${PATH}
|
31 |
-
WORKDIR ${HOME}/app
|
32 |
-
|
33 |
-
RUN curl https://pyenv.run | bash
|
34 |
-
ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
|
35 |
-
ARG PYTHON_VERSION=3.10.11
|
36 |
-
RUN pyenv install ${PYTHON_VERSION} && \
|
37 |
-
pyenv global ${PYTHON_VERSION} && \
|
38 |
-
pyenv rehash && \
|
39 |
-
pip install --no-cache-dir -U pip setuptools wheel
|
40 |
-
|
41 |
-
RUN pip install --no-cache-dir -U torch==2.0.1 torchvision==0.15.2
|
42 |
-
COPY --chown=1000 requirements.txt /tmp/requirements.txt
|
43 |
-
RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
|
44 |
-
|
45 |
-
COPY --chown=1000 . ${HOME}/app
|
46 |
-
ENV PYTHONPATH=${HOME}/app \
|
47 |
-
PYTHONUNBUFFERED=1 \
|
48 |
-
GRADIO_ALLOW_FLAGGING=never \
|
49 |
-
GRADIO_NUM_PORTS=1 \
|
50 |
-
GRADIO_SERVER_NAME=0.0.0.0 \
|
51 |
-
GRADIO_THEME=huggingface \
|
52 |
-
SYSTEM=spaces
|
53 |
-
CMD ["python", "app.py"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
README.md
CHANGED
@@ -3,7 +3,8 @@ title: UniDiffuser
|
|
3 |
emoji: 😻
|
4 |
colorFrom: gray
|
5 |
colorTo: green
|
6 |
-
sdk:
|
|
|
7 |
app_file: app.py
|
8 |
pinned: false
|
9 |
license: other
|
|
|
3 |
emoji: 😻
|
4 |
colorFrom: gray
|
5 |
colorTo: green
|
6 |
+
sdk: gradio
|
7 |
+
skk_version: 3.46.0
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
license: other
|
app.py
CHANGED
@@ -8,6 +8,7 @@ import random
|
|
8 |
import gradio as gr
|
9 |
import numpy as np
|
10 |
import PIL.Image
|
|
|
11 |
import torch
|
12 |
from diffusers import UniDiffuserPipeline
|
13 |
|
@@ -32,6 +33,7 @@ if torch.cuda.is_available():
|
|
32 |
pipe.to(device)
|
33 |
|
34 |
|
|
|
35 |
def run(
|
36 |
mode: str,
|
37 |
prompt: str,
|
|
|
8 |
import gradio as gr
|
9 |
import numpy as np
|
10 |
import PIL.Image
|
11 |
+
import spaces
|
12 |
import torch
|
13 |
from diffusers import UniDiffuserPipeline
|
14 |
|
|
|
33 |
pipe.to(device)
|
34 |
|
35 |
|
36 |
+
@spaces.GPU
|
37 |
def run(
|
38 |
mode: str,
|
39 |
prompt: str,
|
requirements.txt
CHANGED
@@ -1,9 +1,7 @@
|
|
1 |
accelerate==0.23.0
|
2 |
diffusers==0.21.4
|
3 |
gradio==3.46.0
|
4 |
-
|
5 |
-
torch==2.0.
|
6 |
-
torchvision==0.15.
|
7 |
transformers==4.34.0
|
8 |
-
triton==2.0.0
|
9 |
-
xformers==0.0.20
|
|
|
1 |
accelerate==0.23.0
|
2 |
diffusers==0.21.4
|
3 |
gradio==3.46.0
|
4 |
+
spaces==0.15.0
|
5 |
+
torch==2.0.0
|
6 |
+
torchvision==0.15.1
|
7 |
transformers==4.34.0
|
|
|
|