docker layout

Dockerfile

@@ -3,6 +3,7 @@ FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
 ARG DEBIAN_FRONTEND=noninteractive
 
 ENV PYTHONUNBUFFERED=1
+ENV NODE_MAJOR=20
 
 RUN apt-get update && apt-get install --no-install-recommends -y \
     build-essential \
@@ -12,11 +13,16 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
     git \
     ffmpeg \
     google-perftools \
+    ca-certificates curl gnupg \
     && apt-get clean && rm -rf /var/lib/apt/lists/*
     
-
 WORKDIR /code
 
+RUN mkdir -p /etc/apt/keyrings 
+RUN curl -fsSL https://deb.nodesource.com/gpgkey/nodesource-repo.gpg.key | gpg --dearmor -o /etc/apt/keyrings/nodesource.gpg
+RUN echo "deb [signed-by=/etc/apt/keyrings/nodesource.gpg] https://deb.nodesource.com/node_${NODE_MAJOR}.x nodistro main" | tee /etc/apt/sources.list.d/nodesource.list > /dev/null
+RUN apt-get update && apt-get install nodejs -y
+
 COPY ./requirements.txt /code/requirements.txt
 
 # Set up a new user named "user" with user ID 1000
@@ -39,6 +45,4 @@ WORKDIR $HOME/app
 COPY --chown=user . $HOME/app
 
 ENV LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
-# CMD ["uvicorn", "app-img2img:app", "--host", "0.0.0.0", "--port", "7860"]
-# CMD ["uvicorn", "app-txt2img:app", "--host", "0.0.0.0", "--port", "7860"]
-CMD ["uvicorn", "app-controlnet:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["./build-run.sh"]

build-run.sh

@@ -8,4 +8,9 @@ else
     echo -e "\033[1;31m\nfrontend build failed\n\033[0m" >&2  exit 1
 fi
 cd ../
-python run.py --reload  --pipeline controlnet
+#check if var PIPELINE is set otherwise get default
+if [ -z ${PIPELINE+x} ]; then
+    PIPELINE="controlnet"
+fi
+echo -e "\033[1;32m\npipeline: $PIPELINE \033[0m"
+python3 run.py --port 7860 --host 0.0.0.0 --pipeline $PIPELINE

frontend/src/lib/components/VideoInput.svelte

@@ -53,7 +53,7 @@
   }
 </script>
 
-<div class="relative overflow-hidden rounded-lg border border-slate-300">
+<div class="relative mx-auto max-w-lg overflow-hidden rounded-lg border border-slate-300">
   <div class="relative z-10 aspect-square w-full object-cover">
     <video
       class="aspect-square w-full object-cover"

frontend/src/routes/+page.svelte

@@ -110,14 +110,14 @@
   {#if pipelineParams}
     <article class="my-3 grid grid-cols-1 gap-3 sm:grid-cols-2">
       {#if isImageMode}
-        <div class="col-start-1">
+        <div class="sm:col-start-1">
           <VideoInput></VideoInput>
         </div>
       {/if}
-      <div class={isImageMode ? 'col-start-2' : 'col-span-2'}>
+      <div class={isImageMode ? 'sm:col-start-2' : 'col-span-2'}>
         <ImagePlayer />
       </div>
-      <div class="col-span-2">
+      <div class="sm:col-span-2">
         <Button on:click={toggleLcmLive} {disabled} classList={'text-lg my-1 p-2'}>
           {#if isLCMRunning}
             Stop

requirements.txt

@@ -1,6 +1,5 @@
 git+https://github.com/huggingface/diffusers@6f1435332bc74e286af5e88014236e4cc712b747
-transformers==4.34.1
-gradio==3.50.2
+transformers==4.35.2
 --extra-index-url https://download.pytorch.org/whl/cu121;
 torch==2.1.0
 fastapi==0.104.1