Skip to content

Commit e92a603

Browse files
authored
fix dreambooth script. (#1017)
make input_args optional
1 parent 1d04e1b commit e92a603

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/dreambooth/train_dreambooth.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
logger = get_logger(__name__)
2727

2828

29-
def parse_args(input_args):
29+
def parse_args(input_args: None):
3030
parser = argparse.ArgumentParser(description="Simple example of a training script.")
3131
parser.add_argument(
3232
"--pretrained_model_name_or_path",

0 commit comments

Comments
 (0)