[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-05-14 04:24:23 +00:00
parent 0dede489d6
commit 89917e247b

View File

@ -1,3 +1,4 @@
import warnings
from typing import List, Optional, Tuple, Union
import torch
@ -21,7 +22,6 @@ from transformers.models.falcon.modeling_falcon import (
build_alibi_tensor,
)
from transformers.utils import logging
import warnings
from colossalai.pipeline.stage_manager import PipelineStageManager
from colossalai.shardformer.shard import ShardConfig
@ -322,7 +322,7 @@ class FalconPipelineForwards:
hidden_states = outputs[0]
if use_cache is True:
next_decoder_cache = outputs[1]
outputs[1]
if output_attentions:
all_self_attentions = all_self_attentions + (outputs[2 if use_cache else 1],)