From 515f0f1fdc9a76bf63bd746c291dcfec7fc747fb Mon Sep 17 00:00:00 2001 From: Volpeon Date: Thu, 13 Oct 2022 21:11:53 +0200 Subject: Added support for Aesthetic Gradients --- dreambooth.py | 10 ---------- 1 file changed, 10 deletions(-) (limited to 'dreambooth.py') diff --git a/dreambooth.py b/dreambooth.py index 699313e..072142e 100644 --- a/dreambooth.py +++ b/dreambooth.py @@ -215,12 +215,6 @@ def parse_args(): "and an Nvidia Ampere GPU." ), ) - parser.add_argument( - "--local_rank", - type=int, - default=-1, - help="For distributed training: local_rank" - ) parser.add_argument( "--sample_frequency", type=int, @@ -287,10 +281,6 @@ def parse_args(): args = parser.parse_args( namespace=argparse.Namespace(**json.load(f)["args"])) - env_local_rank = int(os.environ.get("LOCAL_RANK", -1)) - if env_local_rank != -1 and env_local_rank != args.local_rank: - args.local_rank = env_local_rank - if args.train_data_file is None: raise ValueError("You must specify --train_data_file") -- cgit v1.2.3-54-g00ecf