Re add BF16 for VAE

2023-12-15 00:33:25 +01:00
parent 169a9aea03
commit 3782b16606
2 changed files with 3 additions and 1 deletions
@@ -12,7 +12,7 @@ class ExtraVAELoader:
 			"required": {
 				"vae_name": (folder_paths.get_filename_list("vae"),),
 				"vae_type": (list(vae_conf.keys()), {"default":"kl-f8"}),
-				"dtype"   : (dtype_list_short,),
+				"dtype"   : ([*dtype_list_short, "BF16"],),
 			}
 		}
 	RETURN_TYPES = ("VAE",)