Spaces:
Sleeping
Sleeping
Commit
•
6b83b21
1
Parent(s):
8fb55f8
forgot arg
Browse files- demo_watermark.py +6 -0
demo_watermark.py
CHANGED
@@ -154,6 +154,12 @@ def parse_args():
|
|
154 |
default=False,
|
155 |
help="Skip the model loading to debug the interface.",
|
156 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
157 |
args = parser.parse_args()
|
158 |
return args
|
159 |
|
|
|
154 |
default=False,
|
155 |
help="Skip the model loading to debug the interface.",
|
156 |
)
|
157 |
+
parser.add_argument(
|
158 |
+
"--seed_separately",
|
159 |
+
type=str2bool,
|
160 |
+
default=True,
|
161 |
+
help="Whether to call the torch seed function before both the unwatermarked and watermarked generate calls.",
|
162 |
+
)
|
163 |
args = parser.parse_args()
|
164 |
return args
|
165 |
|