Remove installation of maxdifussion as part of the dependency image b…

…uild process (#107)

maxdiffusion_dependencies.Dockerfile

@@ -45,7 +45,7 @@ ARG JAX_VERSION
 ENV ENV_JAX_VERSION=$JAX_VERSION
 
 # Set the working directory in the container
-WORKDIR /deps
+WORKDIR /app
 
 # Copy all files from local workspace into docker container
 COPY . .
@@ -54,8 +54,5 @@ RUN echo "Running command: bash setup.sh MODE=$ENV_MODE JAX_VERSION=$ENV_JAX_VER
 
 RUN --mount=type=cache,target=/root/.cache/pip bash setup.sh MODE=${ENV_MODE} JAX_VERSION=${ENV_JAX_VERSION}
 
-# Install MaxDiffusion
-RUN pip install .
-
 # Cleanup
 RUN rm -rf /root/.cache/pip

maxdiffusion_jax_stable_stack_tpu.Dockerfile

@@ -7,19 +7,16 @@ ARG COMMIT_HASH
 
 ENV COMMIT_HASH=$COMMIT_HASH
 
-RUN mkdir -p /deps
+RUN mkdir -p /app
 
 # Set the working directory in the container
-WORKDIR /deps
+WORKDIR /app
 
 # Copy all files from local workspace into docker container
 COPY . .
 
 # Install Maxdiffusion jax stable stack requirements
-RUN pip install -r /deps/requirements_with_jax_stable_stack.txt
-
-# Install MaxDiffusion
-RUN pip install .
+RUN pip install -r /app/requirements_with_jax_stable_stack.txt
 
 # Run the script available in JAX-Stable-Stack base image to generate the manifest file
 RUN bash /jax-stable-stack/generate_manifest.sh PREFIX=maxdiffusion COMMIT_HASH=$COMMIT_HASH

maxdiffusion_runner.Dockerfile

@@ -1,12 +1,10 @@
 ARG BASEIMAGE=maxdiffusion_base_image
 FROM $BASEIMAGE
 
-#FROM maxdiffusion_base_image
-
 # Set the working directory in the container
-WORKDIR /deps
+WORKDIR /app
 
 # Copy all files from local workspace into docker container
 COPY . .
 
-WORKDIR /deps
+WORKDIR /app