"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "0c9c8472e65f630fb2a8f389ea5f875f699a7dbc"
Unverified Commit 07c2607d authored by holazzer's avatar holazzer Committed by GitHub
Browse files

fix use_cache value assign (#13532)

fix use_cache value assign
parent 010965dc
......@@ -1812,7 +1812,7 @@ class ProphetNetModel(ProphetNetPreTrainedModel):
>>> last_hidden_states = outputs.last_hidden_state # main stream hidden states
>>> last_hidden_states_ngram = outputs.last_hidden_state_ngram # predict hidden states
"""
use_cache == use_cache if use_cache is not None else self.config.use_cache
use_cache = use_cache if use_cache is not None else self.config.use_cache
output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
output_hidden_states = (
output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states
......
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