dtype cleanup

This commit is contained in:
City
2023-12-22 15:54:16 +01:00
parent 1a968d7bc8
commit 451c7588c7
4 changed files with 24 additions and 80 deletions
+9 -2
View File
@@ -3,7 +3,14 @@ import folder_paths
from .conf import vae_conf
from .loader import EXVAE
from ..utils.dtype import string_to_dtype, dtype_list_short
from ..utils.dtype import string_to_dtype
dtypes = [
"auto",
"FP32",
"FP16",
"BF16"
]
class ExtraVAELoader:
@classmethod
@@ -12,7 +19,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, "BF16"],),
"dtype" : (dtypes,),
}
}
RETURN_TYPES = ("VAE",)