Merge 31b57a0569 into e57130e9e7

[fix] dll
2026-05-19 17:47:40 +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
100apps	fa71da3bbf	Merge `31b57a0569` into `e57130e9e7`	2026-05-19 17:47:40 +08:00
jingyaogong	e57130e9e7	[fix] dll	2026-05-19 17:43:47 +08:00