Merge 2a65e6a675 into e57130e9e7

[fix] dll
2026-05-21 16:26:09 +08:00 · 2026-05-19 17:43:47 +08:00
2 changed files with 2 additions and 0 deletions
--- a/trainer/train_pretrain_vlm.py
+++ b/trainer/train_pretrain_vlm.py
@ -7,6 +7,7 @@ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 import argparse
 import time
 import warnings
+import datasets
 import torch
 import torch.distributed as dist
 from contextlib import nullcontext
--- a/trainer/train_sft_vlm.py
+++ b/trainer/train_sft_vlm.py
@ -7,6 +7,7 @@ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 import argparse
 import time
 import warnings
+import datasets
 import torch
 import torch.distributed as dist
 from contextlib import nullcontext
Author	SHA1	Message	Date
dynorth	db165dfcad	Merge `2a65e6a675` into `e57130e9e7`	2026-05-21 16:26:09 +08:00
jingyaogong	e57130e9e7	[fix] dll	2026-05-19 17:43:47 +08:00