dockerfile

Dockerfile

@@ -16,12 +16,6 @@ WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
-RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
-
-RUN wget https://github.com/chengzeyi/stable-fast/releases/download/v1.0.5/stable_fast-1.0.5+torch222cu121-cp310-cp310-manylinux2014_x86_64.whl \
-    && pip3 install stable_fast-1.0.5+torch222cu121-cp310-cp310-manylinux2014_x86_64.whl \
-    && rm stable_fast-1.0.5+torch222cu121-cp310-cp310-manylinux2014_x86_64.whl
-
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 # Switch to the "user" user
@@ -37,7 +31,7 @@ ENV HOME=/home/user \
     GRADIO_THEME=huggingface \
     SYSTEM=spaces
 
-#RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
 
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app

anime_app.py

@@ -17,8 +17,7 @@ os.system("nvidia-smi")
 print("TORCH_CUDA", torch.cuda.is_available())
 print("CUDA version:", torch.version.cuda)
 
-from sfast.compilers.diffusion_pipeline_compiler import (compile,
- CompilationConfig)
+from sfast.compilers.diffusion_pipeline_compiler import compile, CompilationConfig
 
 import torch
 from diffusers import (

requirements.txt

@@ -1,3 +1,4 @@
+stable-fast @ https://github.com/chengzeyi/stable-fast/releases/download/v1.0.5/stable_fast-1.0.5+torch222cu121-cp310-cp310-manylinux2014_x86_64.whl
 torch==2.2.0
 torchvision==0.17.0
 torchaudio==2.2.0