Upload folder using huggingface_hub

modeling_midashenglm.py

@@ -16,8 +16,7 @@ from transformers.models.qwen2_5_omni.configuration_qwen2_5_omni import (
 from transformers.models.qwen2_5_omni.modeling_qwen2_5_omni import (
     Qwen2_5OmniThinkerTextModel,
 )
-from transformers.utils import LossKwargs, can_return_tuple
-from typing_extensions import Unpack
+from transformers.utils import can_return_tuple
 
 from .configuration_midashenglm import DashengConfig, MiDashengLMConfig
 
@@ -425,7 +424,7 @@ class Qwen25OmniThinkerTextOnlyDecoder(PreTrainedModel, GenerationMixin):
         output_hidden_states: Optional[bool] = None,
         cache_position: Optional[torch.LongTensor] = None,
         labels: Optional[torch.Tensor] = None,
-        **kwargs: Unpack[LossKwargs],
+        **kwargs,
     ) -> Union[Tuple, Qwen25OmniTextModelOutput]:
         if attention_mask is not None and position_ids is None:
             position_ids = (