From eceeebdf9147fa8daa6acaa69088def0f60315a2 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Thu, 27 Oct 2022 15:51:11 +0200 Subject: [PATCH] Update train_dreambooth.py --- 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 9b7e17241d..8b0b20e07b 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: None): +def parse_args(input_args=None): parser = argparse.ArgumentParser(description="Simple example of a training script.") parser.add_argument( "--pretrained_model_name_or_path",