Skip to content

Commit 6ea8828

Browse files
authored
fix reloading of transformer model (#1147)
1 parent e17fcd6 commit 6ea8828

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

merlin/models/tf/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,7 @@
133133
from merlin.models.utils.dependencies import is_transformers_available
134134

135135
if is_transformers_available():
136+
import transformers
136137
from merlin.models.tf.transformers.block import (
137138
AlbertBlock,
138139
BertBlock,

0 commit comments

Comments
 (0)