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