File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 33import itertools
44import math
55import os
6+ import warnings
67from pathlib import Path
78from typing import Optional
89
@@ -255,10 +256,11 @@ def parse_args(input_args=None):
255256 if args .class_prompt is None :
256257 raise ValueError ("You must specify prompt for class images." )
257258 else :
259+ # logger is not available yet
258260 if args .class_data_dir is not None :
259- logger . warning ("You need not use --class_data_dir without --with_prior_preservation." )
261+ warnings . warn ("You need not use --class_data_dir without --with_prior_preservation." )
260262 if args .class_prompt is not None :
261- logger . warning ("You need not use --class_prompt without --with_prior_preservation." )
263+ warnings . warn ("You need not use --class_prompt without --with_prior_preservation." )
262264
263265 return args
264266
You can’t perform that action at this time.
0 commit comments