Skip to content

GitLab

Commit 9bdcba53 authored by thomwolf's avatar thomwolf
Browse files

fix tests

parent f0bf81e1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/modeling_openai_test.py
View file @ 9bdcba53
...@@ -93,7 +93,7 @@ class OpenAIGPTModelTest(unittest.TestCase): ...@@ -93,7 +93,7 @@ class OpenAIGPTModelTest(unittest.TestCase):
if self.use_labels: if self.use_labels:
mc_labels = OpenAIGPTModelTest.ids_tensor([self.batch_size], self.type_sequence_label_size) mc_labels = OpenAIGPTModelTest.ids_tensor([self.batch_size], self.type_sequence_label_size)
lm_labels = OpenAIGPTModelTest.ids_tensor([self.batch_size, self.n_choices, self.seq_length], self.num_labels) lm_labels = OpenAIGPTModelTest.ids_tensor([self.batch_size, self.n_choices, self.seq_length], self.num_labels)
mc_token_ids = OpenAIGPTModelTest.ids_tensor([self.batch_size, self.n_choices], self.seq_length).float() mc_token_ids = OpenAIGPTModelTest.ids_tensor([self.batch_size, self.n_choices], self.seq_length)
config = OpenAIGPTConfig( config = OpenAIGPTConfig(
vocab_size_or_config_json_file=self.vocab_size, vocab_size_or_config_json_file=self.vocab_size,
......
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