diff --git a/agent_scheduler/shared_opts_backup.py b/agent_scheduler/shared_opts_backup.py index 7bd5075..ce728e0 100644 --- 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 index 62f1044..12c65dc 100644 --- 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')