Merge 31b57a0569 into 2b94dcb3d0

2026-05-16 08:36:20 +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