diff --git a/inference.py b/inference.py index 62fcce8..904f605 100644 --- a/inference.py +++ b/inference.py @@ -106,6 +106,7 @@ def parse_args(input_args=None): def run_inference(args): current_datetime = datetime.datetime.now() + current_datetime = str(current_datetime).replace(":", "_") save_path = f"./result/{current_datetime}_lora" if args.plugin_type == "lora" else f"./result/{current_datetime}_controlnet" os.makedirs(save_path) args.save_path = save_path @@ -122,4 +123,4 @@ def run_inference(args): if __name__ == "__main__": args = parse_args() - run_inference(args) \ No newline at end of file + run_inference(args)