fix type-check

This commit is contained in:
caitianchi 2024-08-12 16:47:59 +08:00
parent a945b3ca8b
commit 89d378c76b

View file

@ -19,8 +19,6 @@
import os import os
import math import math
import warnings import warnings
from dataclasses import dataclass
from typing import Optional, Tuple, Union
import numpy as np import numpy as np
import torch import torch
@ -33,7 +31,6 @@ from transformers.activations import ACT2FN
from transformers.modeling_utils import PreTrainedModel from transformers.modeling_utils import PreTrainedModel
from transformers.configuration_utils import PretrainedConfig from transformers.configuration_utils import PretrainedConfig
from transformers.utils import ( from transformers.utils import (
ModelOutput,
logging, logging,
) )
from transformers.utils import logging from transformers.utils import logging