nikravan commited on
Commit
2e972b8
1 Parent(s): cf8f80d

fix colab error

Browse files
Files changed (1) hide show
  1. visual.py +1 -1
visual.py CHANGED
@@ -66,7 +66,7 @@ class Attention(nn.Module):
66
  out = attention_fn_default(
67
  q, k, v
68
  )
69
- output = self.dense(out.transpose(1, 2).view(B, L, -1))
70
  output = self.output_dropout(output)
71
  return output
72
 
 
66
  out = attention_fn_default(
67
  q, k, v
68
  )
69
+ output = self.dense(out.transpose(1, 2).contiguous().view(B, L, -1))
70
  output = self.output_dropout(output)
71
  return output
72