Merge 2a65e6a675 into 2b94dcb3d0

2026-05-14 11:27:57 +08:00
2 changed files with 0 additions and 2 deletions
--- a/trainer/train_pretrain_vlm.py
+++ b/trainer/train_pretrain_vlm.py
@ -7,7 +7,6 @@ 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,7 +7,6 @@ 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