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.
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,
|