fixed issue with embed name, save whole config to dir instead of just process so it can be easily shared. Only make one config, no timesteps

2023-09-09 12:24:08 -06:00
parent be804c9cf5
commit 2128ac1e08
2 changed files with 6 additions and 6 deletions
--- a/jobs/process/BaseSDTrainProcess.py
+++ b/jobs/process/BaseSDTrainProcess.py
@@ -284,8 +284,8 @@ class BaseSDTrainProcess(BaseTrainProcess):
                # change filename to pt if that is set
                if self.embed_config.save_format == "pt":
                    # replace extension
-                    file_path = os.path.splitext(emb_file_path)[0] + ".pt"
-                self.embedding.save(file_path)
+                    emb_file_path = os.path.splitext(emb_file_path)[0] + ".pt"
+                self.embedding.save(emb_file_path)
        else:
            self.sd.save(
                file_path,