fix colab error
Browse files
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 |
|