zhengchong commited on
Commit
26c80cc
Β·
1 Parent(s): 32663b3

feat: Add new argument for p2p base model path in app.py

Browse files

- Introduced a new command-line argument `--p2p_base_model_path` to specify the path to the base model for evaluation, defaulting to "timbrooks/instruct-pix2pix". This enhancement allows for greater flexibility in model selection during runtime.

Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -28,6 +28,14 @@ def parse_args():
28
  "The path to the base model to use for evaluation. This can be a local path or a model identifier from the Model Hub."
29
  ),
30
  )
 
 
 
 
 
 
 
 
31
  parser.add_argument(
32
  "--resume_path",
33
  type=str,
 
28
  "The path to the base model to use for evaluation. This can be a local path or a model identifier from the Model Hub."
29
  ),
30
  )
31
+ parser.add_argument(
32
+ "--p2p_base_model_path",
33
+ type=str,
34
+ default="timbrooks/instruct-pix2pix",
35
+ help=(
36
+ "The path to the base model to use for evaluation. This can be a local path or a model identifier from the Model Hub."
37
+ ),
38
+ )
39
  parser.add_argument(
40
  "--resume_path",
41
  type=str,