From f56b0a9eb7af8f45da7419cea94b9949802135bc Mon Sep 17 00:00:00 2001 From: jingyaogong Date: Tue, 19 May 2026 17:46:38 +0800 Subject: [PATCH] [fix] dll --- trainer/train_sft_omni.py | 1 + trainer/trainer_utils.py | 1 + 2 files changed, 2 insertions(+) diff --git a/trainer/train_sft_omni.py b/trainer/train_sft_omni.py index 9df8c8e..612f5e7 100644 --- a/trainer/train_sft_omni.py +++ b/trainer/train_sft_omni.py @@ -4,6 +4,7 @@ import sys __package__ = "trainer" sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) +import datasets import argparse import time import warnings diff --git a/trainer/trainer_utils.py b/trainer/trainer_utils.py index 59afa41..6789e94 100644 --- a/trainer/trainer_utils.py +++ b/trainer/trainer_utils.py @@ -5,6 +5,7 @@ import os import random import math import numpy as np +import datasets import torch import torch.distributed as dist from torch.utils.data import Sampler