From 4de917db3a1ef5d3003bc03d33f2b624aae5164b Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 14 Jan 2024 19:50:49 -0600
Subject: [PATCH] fix(api): include networks in launch scripts

---
 api/launch.bat | 6 +++---
 api/launch.ps1 | 6 +++---
 api/launch.sh  | 7 ++++---
 3 files changed, 10 insertions(+), 9 deletions(-)

diff --git a/api/launch.bat b/api/launch.bat
index e1bd53e8..2f20a471 100644
--- a/api/launch.bat
+++ b/api/launch.bat
@@ -5,11 +5,11 @@ echo "This launch.bat script is deprecated in favor of launch.ps1 and will be re
 echo "Downloading and converting models to ONNX format..."
 IF "%ONNX_WEB_EXTRA_MODELS%"=="" (set ONNX_WEB_EXTRA_MODELS=..\models\extras.json)
 python -m onnx_web.convert ^
---sources ^
---diffusion ^
---upscaling ^
 --correction ^
+--diffusion ^
 --networks ^
+--sources ^
+--upscaling ^
 --extras=%ONNX_WEB_EXTRA_MODELS% ^
 --token=%HF_TOKEN% %ONNX_WEB_EXTRA_ARGS%
 
diff --git a/api/launch.ps1 b/api/launch.ps1
index 4e378755..420a3554 100644
--- a/api/launch.ps1
+++ b/api/launch.ps1
@@ -3,11 +3,11 @@
 echo "Downloading and converting models to ONNX format..."
 IF ($Env:ONNX_WEB_EXTRA_MODELS -eq "") {$Env:ONNX_WEB_EXTRA_MODELS="..\models\extras.json"}
 python -m onnx_web.convert `
---sources `
---diffusion `
---upscaling `
 --correction `
+--diffusion `
 --networks `
+--sources `
+--upscaling `
 --extras=$Env:ONNX_WEB_EXTRA_MODELS `
 --token=$Env:HF_TOKEN $Env:ONNX_WEB_EXTRA_ARGS
 
diff --git a/api/launch.sh b/api/launch.sh
index 8e793771..e4523bbc 100755
--- a/api/launch.sh
+++ b/api/launch.sh
@@ -17,10 +17,11 @@ fi
 
 echo "Downloading and converting models to ONNX format..."
 python3 -m onnx_web.convert \
-  --sources \
-  --diffusion \
-  --upscaling \
   --correction \
+  --diffusion \
+  --networks \
+  --sources \
+  --upscaling \
   --extras=${ONNX_WEB_EXTRA_MODELS:-../models/extras.json} \
   --token=${HF_TOKEN:-} \
   ${ONNX_WEB_EXTRA_ARGS:-}