From 00bd3d54a3841c0c4944bd025bd998900b8cc28f Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Tue, 13 Aug 2024 17:08:27 -0600
Subject: [PATCH] Actually use the save dtype from the config file.

---
 toolkit/config_modules.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/toolkit/config_modules.py b/toolkit/config_modules.py
index 561d97e8..f7111a32 100644
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -18,7 +18,7 @@ if TYPE_CHECKING:
 class SaveConfig:
     def __init__(self, **kwargs):
         self.save_every: int = kwargs.get('save_every', 1000)
-        self.dtype: str = kwargs.get('save_dtype', 'float16')
+        self.dtype: str = kwargs.get('dtype', 'float16')
         self.max_step_saves_to_keep: int = kwargs.get('max_step_saves_to_keep', 5)
         self.save_format: SaveFormat = kwargs.get('save_format', 'safetensors')
         if self.save_format not in ['safetensors', 'diffusers']: