From b8fccdc8f5ea5c9e09a2de2f76df0082bd3d305a Mon Sep 17 00:00:00 2001
From: Sheldon Hull <sheldonhull@users.noreply.github.com>
Date: Wed, 14 Sep 2022 13:41:34 -0500
Subject: [PATCH] chore: stream realtime output

---
 Taskfile.yml | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/Taskfile.yml b/Taskfile.yml
index a3c1110..a22005a 100644
--- a/Taskfile.yml
+++ b/Taskfile.yml
@@ -1,6 +1,6 @@
 version: '3'
 # silent: true
-output: group
+#output: group
 dotenv: [.env]
 
 # Default environment variables for projects
@@ -67,9 +67,10 @@ tasks:
     interactive: true
     cmds:
     - task: vars
-    - |
-      docker compose --profile {{ .UI }} up --build
     - task: download
+    - |
+      docker compose --profile {{ .UI }} build
+
   download:
     desc: get extras downloaded
     cmds:
@@ -83,6 +84,7 @@ tasks:
       curl --progress-bar -C - https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth -o {{ .MODEL_DIRECTORY }}/RealESRGAN_x2plus.pth
       curl --progress-bar -C - https://heibox.uni-heidelberg.de/f/578df07c8fc04ffbadf3/?dl=1 -o {{ .MODEL_DIRECTORY }}/LDSR.ckpt
       curl --progress-bar -C - https://heibox.uni-heidelberg.de/f/31a76b13ea27482981b4/?dl=1 -o {{ .MODEL_DIRECTORY }}/LDSR.yaml
+      curl --progress-bar -C - https://thisanimedoesnotexist.ai/downloads/wd-v1-2-full-ema.ckpt -o {{ .MODEL_DIRECTORY }}/waifu-diffusion.ckpt
     status:
       - test -f {{ .MODEL_DIRECTORY }}/model.ckpt
       - test -f {{ .MODEL_DIRECTORY }}/GFPGANv1.3.pth
@@ -90,4 +92,6 @@ tasks:
       - test -f {{ .MODEL_DIRECTORY }}/RealESRGAN_x4plus_anime_6B.pth
       - test -f {{ .MODEL_DIRECTORY }}/RealESRGAN_x2plus.pth
       - test -f {{ .MODEL_DIRECTORY }}/LDSR.ckpt
-      - test -f {{ .MODEL_DIRECTORY }}/LDSR.yaml
\ No newline at end of file
+      - test -f {{ .MODEL_DIRECTORY }}/LDSR.yaml
+      - test -f {{ .MODEL_DIRECTORY }}/waifu-diffusion.ckpt
+