diff --git a/core/interact/interact.py b/core/interact/interact.py index 7ad0d1c..92d59ea 100644 --- a/core/interact/interact.py +++ b/core/interact/interact.py @@ -46,7 +46,7 @@ class InteractBase(object): self.process_messages_callbacks = {} self.default_answers = {} - answer_filename = '../../../worspace/interact/interact_dict.pkl' + answer_filename = 'worspace/interact/interact_dict.pkl' if os.path.exists(answer_filename): with open(answer_filename, 'rb') as file: self.default_answers = pickle.load(file) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index cf4bc97..3a6930a 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -85,7 +85,7 @@ def trainerThread (s2c, c2s, e, if is_reached_goal: io.log_info('Model already trained to target iteration. You can use preview.') - answer_filename = '../../../worspace/interact/interact_dict.pkl' + answer_filename = 'DeepFaceLab_Linux/worspace/interact/interact_dict.pkl' if os.path.exists(answer_filename): with open(answer_filename, 'rb') as file: parameters = pickle.load(file) @@ -171,6 +171,7 @@ def trainerThread (s2c, c2s, e, model_save() is_reached_goal = True io.log_info ('You can use preview now.') + answer_filename = 'DeepFaceLab_Linux/worspace/interact/interact_dict.pkl' if os.path.exists(answer_filename): with open(answer_filename, 'rb') as file: parameters = pickle.load(file) diff --git a/models/ModelBase.py b/models/ModelBase.py index 2a4f755..f3a300a 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -295,6 +295,7 @@ class ModelBase(object): self.choose_preview_history = io.input_bool("Randomly choose new image for preview history", False, help_message="Preview image history will stay stuck with old faces if you reuse the same model on different celebs. Choose no unless you are changing src/dst to a new person") def ask_target_iter(self, default_value=0): + answer_filename = 'DeepFaceLab_Linux/worspace/interact/interact_dict.pkl' if os.path.exists(answer_filename): with open(answer_filename, 'rb') as file: parameters = pickle.load(file)