From 951dbc94a54d47d63948347d98c68f32a014e941 Mon Sep 17 00:00:00 2001
From: Kasisnu <5044120+kasisnu@users.noreply.github.com>
Date: Mon, 22 Apr 2024 18:54:23 +0530
Subject: [PATCH] smaller tt2 images

---
 tts/tacotron2_v1_early_fakeyou/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tts/tacotron2_v1_early_fakeyou/Dockerfile b/tts/tacotron2_v1_early_fakeyou/Dockerfile
index 5ec522a..9b3806d 100644
--- a/tts/tacotron2_v1_early_fakeyou/Dockerfile
+++ b/tts/tacotron2_v1_early_fakeyou/Dockerfile
@@ -106,7 +106,7 @@ WORKDIR models/tts
 # NB: Setuptools fix: https://github.com/tensorflow/tensorflow/issues/34302#issuecomment-554450289
 RUN python3.6 -m pip install --upgrade pip setuptools \
   && python3.6 -m pip install -r requirements-tacotron-python36.txt \
-  && python3.6 -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 
+  && python3.6 -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 && python3.6 -m pip cache purge
 
 # Final image
 #  FROM ubuntu:xenial
@@ -140,7 +140,7 @@ RUN apt-get install -y libespeak-ng1
 RUN python3 --version
 
 # TODO: Use venv to do this instead.
-RUN pip3 install gdown
+RUN pip3 install gdown && pip3 cache purge
 
 # NB(echelon,2022-07-14): Run a newer version of pytorch which will support RTX3090.
 #  Fixes bug, "NVIDIA GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation."