refactor(khaosz/trainer): 移除未使用的导入模块

This commit is contained in:
ViperEkura 2025-10-04 21:45:53 +08:00
parent 9d5aa952e0
commit fc98d9b7e6
1 changed files with 2 additions and 4 deletions

View File

@ -5,11 +5,9 @@ import torch.nn as nn
import torch.nn.functional as F
from torch import Tensor
from torch.optim import Optimizer
from torch.utils.data import Dataset
from typing import Any, Literal, Optional, Tuple, Callable, Dict
from typing import Any, Literal, Tuple, Callable, Dict
from abc import ABC, abstractmethod
from dataclasses import asdict, dataclass, field
from dataclasses import dataclass, field
def get_logprobs(model:nn.Module, input_ids: Tensor, mask: Tensor, pad_token_id: int):