github
/
minimind-o
mirror of https://github.com/jingyaogong/minimind-o.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[fix] dll

pull/21/head
jingyaogong 2026-05-19 17:46:38 +08:00
parent ca060fbed8
commit f56b0a9eb7
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
trainer/train_sft_omni.py
View File

@ -4,6 +4,7 @@ import sys
__package__ = "trainer" __package__ = "trainer"
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
import datasets
import argparse import argparse
import time import time
import warnings import warnings

1
trainer/trainer_utils.py
View File

@ -5,6 +5,7 @@ import os
import random import random
import math import math
import numpy as np import numpy as np
import datasets
import torch import torch
import torch.distributed as dist import torch.distributed as dist
from torch.utils.data import Sampler from torch.utils.data import Sampler