SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[GPTNeoX] Nit in config (#24349)

Browse Source 
* add raise value error for attention size

* nits to fix test_config

* style
This commit is contained in:
Arthur
2023-06-20 19:19:19 +02:00
committed by GitHub
parent c2882403c4
commit e5c760d636
3 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/gpt_neox/test_modeling_gpt_neox.py
View File

@@ -253,7 +253,7 @@ class GPTNeoXModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi
def setUp(self):
self.model_tester = GPTNeoXModelTester(self)
self.config_tester = ConfigTester(self, config_class=GPTNeoXConfig, hidden_size=37)
self.config_tester = ConfigTester(self, config_class=GPTNeoXConfig, hidden_size=64, num_attention_heads=8)
def test_config(self):
self.config_tester.run_common_tests()