Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "782cce7c44bb6d60e4d83d911c1bae44350acb25"
Commit 568c0ffb authored by thomwolf's avatar thomwolf
Browse files

adding T5 model

parent 60a5babd
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 412 additions and 63 deletions
transformers/modeling_encoder_decoder.py
View file @ 568c0ffb
......@@ -217,9 +217,7 @@ class PreTrainedEncoderDecoder(nn.Module):
encoder_hidden_states = kwargs_encoder.pop("hidden_states", None)
if encoder_hidden_states is None:
encoder_outputs = self.encoder(encoder_input_ids, **kwargs_encoder)
encoder_hidden_states = encoder_outputs[
0
] # output the last layer hidden state
encoder_hidden_states = encoder_outputs[0]
else:
encoder_outputs = ()
......
transformers/modeling_t5.py
View file @ 568c0ffb
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment