nodes/smthemex-ComfyUI_SongGeneration
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix new transformer error

Browse Source
This commit is contained in:
smthemex
2025-10-17 20:48:37 +08:00
committed by GitHub
parent fbf709be0b
commit 08c04126cf
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SongGeneration/codeclm/tokenizer/Flow1dVAE/models_gpt/models/gpt2_rope2_time_new_correct_mask_noncasual_reflow.py
+5 -1
View File
@@ -37,7 +37,11 @@ from transformers.modeling_outputs import (
SequenceClassifierOutputWithPast,
TokenClassifierOutput,
)
from transformers.modeling_utils import PreTrainedModel, SequenceSummary
from transformers.modeling_utils import PreTrainedModel
try:
from transformers.models.gpt2.modeling_gpt2 import GPT2SequenceSummary as SequenceSummary
except:
from transformers.modeling_utils import SequenceSummary
from transformers.pytorch_utils import Conv1D, find_pruneable_heads_and_indices, prune_conv1d_layer
from transformers.utils import (
ModelOutput,