Commit 73f2c342 authored by thomwolf's avatar thomwolf
Browse files

fixing template

parent 3835e1e6
......@@ -84,7 +84,7 @@ class XxxConfig(PretrainedConfig):
summary_first_dropout=0.1,
**kwargs):
super(XxxConfig, self).__init__(**kwargs)
self.vocab_size = vocab_size_or_config_json_file if isinstance(vocab_size_or_config_json_file, six.string_types) else -1
self.vocab_size = vocab_size_or_config_json_file if isinstance(vocab_size_or_config_json_file, int) else -1
self.n_ctx = n_ctx
self.n_positions = n_positions
self.n_embd = n_embd
......
......@@ -280,7 +280,6 @@ class XxxModel(XxxPreTrainedModel):
self.init_weights()
@property
def get_input_embeddings(self):
return self.embeddings.word_embeddings
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment