Skip to content
Snippets Groups Projects

Fixed multiword prediction + bug that made the code write empty predictions

Merged Martyna Wiącek requested to merge multiword_fix into main
Viewing commit 5fae577a
Show latest version
2 files
+ 5
5
Compare changes
  • Side-by-side
  • Inline
Files
2
+ 1
1
@@ -39,7 +39,7 @@ class ReLUActivation(Activation):
@Registry.register('gelu')
class ReLUActivation(Activation):
class GELUActivation(Activation):
def __init__(self):
super().__init__()
self.__torch_activation = torch.nn.GELU()