From 18cbab1d8494112492eff713575c13d1ed0b261d Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Fri, 17 Nov 2023 15:09:02 +0000
Subject: [PATCH] Add VAE model

---
 tests-ui/setup.js | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests-ui/setup.js b/tests-ui/setup.js
index 0f368ab22..8bbd9dcdf 100644
--- a/tests-ui/setup.js
+++ b/tests-ui/setup.js
@@ -20,6 +20,7 @@ async function setup() {
 							// Modify the response data to add some checkpoints
 							const objectInfo = JSON.parse(data);
 							objectInfo.CheckpointLoaderSimple.input.required.ckpt_name[0] = ["model1.safetensors", "model2.ckpt"];
+							objectInfo.VAELoader.input.required.vae_name[0] = ["vae1.safetensors", "vae2.ckpt"];
 
 							data = JSON.stringify(objectInfo, undefined, "\t");