JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2026-06-02 17:48:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge 91960586228e46a4881cf076a7a3027a84a49915 into 7a1af7154511e0ce4e4be8d62faa8c5e5a3532d2

Browse Source
This commit is contained in:
Harikrishna KP 2026-02-12 00:06:12 +05:30 committed by GitHub
commit f3ede921ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
finetune/trainer.py
View File

@ -588,9 +588,9 @@ class Trainer:
"image": {"type": "image", "value": image}, "image": {"type": "image", "value": image},
"video": {"type": "video", "value": video}, "video": {"type": "video", "value": video},
} }
for i, (artifact_type, artifact_value) in enumerate(validation_artifacts): for artifact_idx, (artifact_type, artifact_value) in enumerate(validation_artifacts):
artifacts.update( artifacts.update(
{f"artifact_{i}": {"type": artifact_type, "value": artifact_value}} {f"artifact_{artifact_idx}": {"type": artifact_type, "value": artifact_value}}
) )
logger.debug( logger.debug(
f"Validation artifacts on process {accelerator.process_index}: {list(artifacts.keys())}", f"Validation artifacts on process {accelerator.process_index}: {list(artifacts.keys())}",