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

fixing template

parent 3835e1e6
...@@ -84,7 +84,7 @@ class XxxConfig(PretrainedConfig): ...@@ -84,7 +84,7 @@ class XxxConfig(PretrainedConfig):
summary_first_dropout=0.1, summary_first_dropout=0.1,
**kwargs): **kwargs):
super(XxxConfig, self).__init__(**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_ctx = n_ctx
self.n_positions = n_positions self.n_positions = n_positions
self.n_embd = n_embd self.n_embd = n_embd
......
...@@ -280,7 +280,6 @@ class XxxModel(XxxPreTrainedModel): ...@@ -280,7 +280,6 @@ class XxxModel(XxxPreTrainedModel):
self.init_weights() self.init_weights()
@property
def get_input_embeddings(self): def get_input_embeddings(self):
return self.embeddings.word_embeddings 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