Unverified Commit 79838446 authored by orgoro's avatar orgoro Committed by GitHub
Browse files

bugfix: div-->dim (#18135)

parent e630dad5
...@@ -2181,7 +2181,7 @@ class PerceiverBasicDecoder(PerceiverAbstractDecoder): ...@@ -2181,7 +2181,7 @@ class PerceiverBasicDecoder(PerceiverAbstractDecoder):
if self.concat_preprocessed_input: if self.concat_preprocessed_input:
if inputs_without_pos is None: if inputs_without_pos is None:
raise ValueError("Value is required for inputs_without_pos if concat_preprocessed_input is True") raise ValueError("Value is required for inputs_without_pos if concat_preprocessed_input is True")
pos_emb = torch.cat([inputs_without_pos, pos_emb], div=-1) pos_emb = torch.cat([inputs_without_pos, pos_emb], dim=-1)
return pos_emb return pos_emb
......
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