From e92a603cab5c2c75c8d7fb348b9edb37deb24018 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Thu, 27 Oct 2022 11:44:06 +0200 Subject: [PATCH] fix dreambooth script. (#1017) make input_args optional --- examples/dreambooth/train_dreambooth.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py index 6d3880b1a7..d3720f0cad 100644 --- a/examples/dreambooth/train_dreambooth.py +++ b/examples/dreambooth/train_dreambooth.py @@ -26,7 +26,7 @@ from transformers import CLIPTextModel, CLIPTokenizer logger = get_logger(__name__) -def parse_args(input_args): +def parse_args(input_args: None): parser = argparse.ArgumentParser(description="Simple example of a training script.") parser.add_argument( "--pretrained_model_name_or_path",