fix: outdir_path_grids_old

bluelovers · Jun 4, 2024 · fcab048 · fcab048
1 parent fcaf117
commit fcab048
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/agent_scheduler/shared_opts_backup.py b/agent_scheduler/shared_opts_backup.py
@@ -50,10 +50,10 @@ def set_shared_opts_core(self, key: str, value):
         """
         if not self.is_backup_exists(key):
             old = getattr(self.shared_opts, key, None)
-            log.info(f"[AgentScheduler] {key} is backup: {old}")
             self.backup[key] = old
+            print(f"[AgentScheduler] {key} is backup: {old}")
         self.shared_opts.set(key, value)
-        log.info(f"[AgentScheduler] {key} is changed: {value}")
+        print(f"[AgentScheduler] {key} is changed: {value}")
 
     def set_shared_opts(self, **kwargs):
         """
@@ -79,3 +79,4 @@ def restore_shared_opts(self):
         """
         for attr, value in self.backup.items():
             self.shared_opts.set(attr, value)
+            print(f"[AgentScheduler] {attr} is restore: {value}")
diff --git a/agent_scheduler/task_runner.py b/agent_scheduler/task_runner.py
@@ -359,7 +359,7 @@ def change_output_dir():
                         key_grids = "outdir_txt2img_grids"
 
                     outdir_path_samples_old = Path(shared_opts_backup.get_backup_value(key_samples))
-                    outdir_path_grids_old = Path(shared_opts_backup.get_backup_value(key_samples))
+                    outdir_path_grids_old = Path(shared_opts_backup.get_backup_value(key_grids))
 
                     outdir_path_root = outdir_path_samples_old.joinpath('..', 'agent-scheduler')