From 62b84531f4fb647500514093f1676dddf282b2ed Mon Sep 17 00:00:00 2001
From: City <125218114+city96@users.noreply.github.com>
Date: Tue, 2 Jul 2024 23:02:31 +0200
Subject: [PATCH] Add ostris/vae-kl-f8-d16

https://huggingface.co/ostris/vae-kl-f8-d16
---
 VAE/conf.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/VAE/conf.py b/VAE/conf.py
index ee4b256..87112dd 100644
--- a/VAE/conf.py
+++ b/VAE/conf.py
@@ -31,6 +31,20 @@ vae_conf = {
 		"num_res_blocks"   : 2,
 		"attn_resolutions" : [],
 	},
+	"kl-f8-d16": { # 16 channel VAE from https://huggingface.co/ostris/vae-kl-f8-d16/tree/main
+		"type"             : "AutoencoderKL",
+		"embed_scale"      : 8,
+		"embed_dim"        : 16,
+		"z_channels"       : 16,
+		"double_z"         : True,
+		"resolution"       : 256,
+		"in_channels"      : 3,
+		"out_ch"           : 3,
+		"ch"               : 128, 
+		"ch_mult"          : [1,1,2,4],
+		"num_res_blocks"   : 2,
+		"attn_resolutions" : [],
+	},
 	"kl-f16": {
 		"type"             : "AutoencoderKL",
 		"embed_scale"      : 16,