We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b2e797b commit 84b118bCopy full SHA for 84b118b
1 file changed
torchTextClassifiers/torchTextClassifiers.py
@@ -125,10 +125,11 @@ def __init__(
125
self.model_config = model_config
126
self.tokenizer = tokenizer
127
128
- if not self.tokenizer.trained:
129
- raise RuntimeError(
130
- f"Tokenizer {type(self.tokenizer)} must be trained before initializing the classifier."
131
- )
+ if hasattr(self.tokenizer, "trained"):
+ if not self.tokenizer.trained:
+ raise RuntimeError(
+ f"Tokenizer {type(self.tokenizer)} must be trained before initializing the classifier."
132
+ )
133
134
self.vocab_size = tokenizer.vocab_size
135
self.embedding_dim = model_config.embedding_dim
0 commit comments