[fix] dll

2026-05-19 17:43:47 +08:00 · 2026-05-19 17:43:47 +08:00 · e57130e9e7
parent 2b94dcb3d0
commit e57130e9e7
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