zai-org/CogVideo
1
0
Fork 0
mirror of https://github.com/zai-org/CogVideo.git synced 2026-10-12 02:27:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct do_validation argument parsing

Browse Source
This commit is contained in:
OleehyO
2025-01-13 12:48:21 +00:00
parent 455b44a7b5
commit 4878edd0cf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetune/schemas/args.py
+1 -1
View File
@@ -229,7 +229,7 @@ class Args(BaseModel):
parser.add_argument("--resume_from_checkpoint", type=str, default=None)
# Validation
parser.add_argument("--do_validation", type=bool, default=False)
parser.add_argument("--do_validation", type=lambda x: x.lower() == 'true', default=False)
parser.add_argument("--validation_steps", type=int, default=None)
parser.add_argument("--validation_dir", type=str, default=None)
parser.add_argument("--validation_prompts", type=str, default=None)