-
Notifications
You must be signed in to change notification settings - Fork 486
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Split modeling tests to separate files and CI jobs
- Loading branch information
Github Executorch
committed
Dec 18, 2024
1 parent
c36e2e2
commit f0c76d4
Showing
7 changed files
with
316 additions
and
143 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,56 @@ | ||
| # coding=utf-8 | ||
| # Copyright 2024 The HuggingFace Team. All rights reserved. | ||
| # | ||
| # Licensed under the Apache License, Version 2.0 (the "License"); | ||
| # you may not use this file except in compliance with the License. | ||
| # You may obtain a copy of the License at | ||
| # | ||
| # http://www.apache.org/licenses/LICENSE-2.0 | ||
| # | ||
| # Unless required by applicable law or agreed to in writing, software | ||
| # distributed under the License is distributed on an "AS IS" BASIS, | ||
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| # See the License for the specific language governing permissions and | ||
| # limitations under the License. | ||
|
|
||
| import os | ||
| import tempfile | ||
| import unittest | ||
|
|
||
| import pytest | ||
| from executorch.extension.pybindings.portable_lib import ExecuTorchModule | ||
| from transformers import AutoTokenizer | ||
| from transformers.testing_utils import ( | ||
| slow, | ||
| ) | ||
|
|
||
| from optimum.executorchruntime import ExecuTorchModelForCausalLM | ||
|
|
||
|
|
||
| class ExecuTorchModelIntegrationTest(unittest.TestCase): | ||
| def __init__(self, *args, **kwargs): | ||
| super().__init__(*args, **kwargs) | ||
|
|
||
| @slow | ||
| @pytest.mark.run_slow | ||
| def test_gemma_text_generation_with_xnnpack(self): | ||
| # TODO: Swithc to use google/gemma-2b once https://github.com/huggingface/optimum/issues/2127 is fixed | ||
| # model_id = "google/gemma-2b" | ||
| model_id = "weqweasdas/RM-Gemma-2B" | ||
| model = ExecuTorchModelForCausalLM.from_pretrained( | ||
| model_name_or_path=model_id, | ||
| export=True, | ||
| task="text-generation", | ||
| recipe="xnnpack", | ||
| ) | ||
| self.assertIsInstance(model, ExecuTorchModelForCausalLM) | ||
| self.assertIsInstance(model.model, ExecuTorchModule) | ||
|
|
||
| EXPECTED_GENERATED_TEXT = "Hello I am doing a project for my school and I need to write a report on the history of the United States." | ||
| tokenizer = AutoTokenizer.from_pretrained(model_id) | ||
| generated_text = model.text_generation( | ||
| tokenizer=tokenizer, | ||
| prompt="Hello I am doing a project for my school", | ||
| max_seq_len=len(tokenizer.encode(EXPECTED_GENERATED_TEXT)), | ||
| ) | ||
| self.assertEqual(generated_text, EXPECTED_GENERATED_TEXT) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,58 @@ | ||
| # coding=utf-8 | ||
| # Copyright 2024 The HuggingFace Team. All rights reserved. | ||
| # | ||
| # Licensed under the Apache License, Version 2.0 (the "License"); | ||
| # you may not use this file except in compliance with the License. | ||
| # You may obtain a copy of the License at | ||
| # | ||
| # http://www.apache.org/licenses/LICENSE-2.0 | ||
| # | ||
| # Unless required by applicable law or agreed to in writing, software | ||
| # distributed under the License is distributed on an "AS IS" BASIS, | ||
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| # See the License for the specific language governing permissions and | ||
| # limitations under the License. | ||
|
|
||
| import os | ||
| import tempfile | ||
| import unittest | ||
|
|
||
| import pytest | ||
| from executorch.extension.pybindings.portable_lib import ExecuTorchModule | ||
| from transformers import AutoTokenizer | ||
| from transformers.testing_utils import ( | ||
| slow, | ||
| ) | ||
|
|
||
| from optimum.executorchruntime import ExecuTorchModelForCausalLM | ||
|
|
||
|
|
||
| class ExecuTorchModelIntegrationTest(unittest.TestCase): | ||
| def __init__(self, *args, **kwargs): | ||
| super().__init__(*args, **kwargs) | ||
|
|
||
| @slow | ||
| @pytest.mark.run_slow | ||
| def test_gemma2_text_generation_with_xnnpack(self): | ||
| # TODO: Swithc to use google/gemma-2-2b once https://github.com/huggingface/optimum/issues/2127 is fixed | ||
| # model_id = "google/gemma-2-2b" | ||
| model_id = "unsloth/gemma-2-2b-it" | ||
| model = ExecuTorchModelForCausalLM.from_pretrained( | ||
| model_name_or_path=model_id, | ||
| export=True, | ||
| task="text-generation", | ||
| recipe="xnnpack", | ||
| ) | ||
| self.assertIsInstance(model, ExecuTorchModelForCausalLM) | ||
| self.assertIsInstance(model.model, ExecuTorchModule) | ||
|
|
||
| EXPECTED_GENERATED_TEXT = ( | ||
| "Hello I am doing a project for my school and I need to make sure it is a great to be creative and I can!" | ||
| ) | ||
| tokenizer = AutoTokenizer.from_pretrained(model_id) | ||
| generated_text = model.text_generation( | ||
| tokenizer=tokenizer, | ||
| prompt="Hello I am doing a project for my school", | ||
| max_seq_len=len(tokenizer.encode(EXPECTED_GENERATED_TEXT)), | ||
| ) | ||
| self.assertEqual(generated_text, EXPECTED_GENERATED_TEXT) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,84 @@ | ||
| # coding=utf-8 | ||
| # Copyright 2024 The HuggingFace Team. All rights reserved. | ||
| # | ||
| # Licensed under the Apache License, Version 2.0 (the "License"); | ||
| # you may not use this file except in compliance with the License. | ||
| # You may obtain a copy of the License at | ||
| # | ||
| # http://www.apache.org/licenses/LICENSE-2.0 | ||
| # | ||
| # Unless required by applicable law or agreed to in writing, software | ||
| # distributed under the License is distributed on an "AS IS" BASIS, | ||
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| # See the License for the specific language governing permissions and | ||
| # limitations under the License. | ||
|
|
||
| import os | ||
| import tempfile | ||
| import unittest | ||
|
|
||
| import pytest | ||
| from executorch.extension.pybindings.portable_lib import ExecuTorchModule | ||
| from transformers import AutoTokenizer | ||
| from transformers.testing_utils import ( | ||
| slow, | ||
| ) | ||
|
|
||
| from optimum.executorchruntime import ExecuTorchModelForCausalLM | ||
|
|
||
|
|
||
| class ExecuTorchModelIntegrationTest(unittest.TestCase): | ||
| def __init__(self, *args, **kwargs): | ||
| super().__init__(*args, **kwargs) | ||
|
|
||
| @slow | ||
| @pytest.mark.run_slow | ||
| def test_llama3_2_1b_text_generation_with_xnnpack(self): | ||
| # TODO: Swithc to use meta-llama/Llama-3.2-1B once https://github.com/huggingface/optimum/issues/2127 is fixed | ||
| # model_id = "lama/Llama-3.2-1B" | ||
| model_id = "NousResearch/Llama-3.2-1B" | ||
| model = ExecuTorchModelForCausalLM.from_pretrained( | ||
| model_name_or_path=model_id, | ||
| export=True, | ||
| task="text-generation", | ||
| recipe="xnnpack", | ||
| ) | ||
| self.assertIsInstance(model, ExecuTorchModelForCausalLM) | ||
| self.assertIsInstance(model.model, ExecuTorchModule) | ||
|
|
||
| EXPECTED_GENERATED_TEXT = "Simply put, the theory of relativity states that the laws of physics are the same in all inertial frames of reference." | ||
| tokenizer = AutoTokenizer.from_pretrained(model_id) | ||
| generated_text = model.text_generation( | ||
| tokenizer=tokenizer, | ||
| prompt="Simply put, the theory of relativity states that", | ||
| max_seq_len=len(tokenizer.encode(EXPECTED_GENERATED_TEXT)), | ||
| ) | ||
| self.assertEqual(generated_text, EXPECTED_GENERATED_TEXT) | ||
|
|
||
| @slow | ||
| @pytest.mark.run_slow | ||
| def test_llama3_2_3b_text_generation_with_xnnpack(self): | ||
| # TODO: Swithc to use meta-llama/Llama-3.2-3B once https://github.com/huggingface/optimum/issues/2127 is fixed | ||
| # model_id = "lama/Llama-3.2-3B" | ||
| model_id = "NousResearch/Hermes-3-Llama-3.2-3B" | ||
| model = ExecuTorchModelForCausalLM.from_pretrained( | ||
| model_name_or_path=model_id, | ||
| export=True, | ||
| task="text-generation", | ||
| recipe="xnnpack", | ||
| ) | ||
| self.assertIsInstance(model, ExecuTorchModelForCausalLM) | ||
| self.assertIsInstance(model.model, ExecuTorchModule) | ||
|
|
||
| EXPECTED_GENERATED_TEXT = ( | ||
| "Simply put, the theory of relativity states that time is relative and can be affected " | ||
| "by an object's speed. This theory was developed by Albert Einstein in the early 20th " | ||
| "century. The theory has two parts" | ||
| ) | ||
| tokenizer = AutoTokenizer.from_pretrained(model_id) | ||
| generated_text = model.text_generation( | ||
| tokenizer=tokenizer, | ||
| prompt="Simply put, the theory of relativity states that", | ||
| max_seq_len=len(tokenizer.encode(EXPECTED_GENERATED_TEXT)), | ||
| ) | ||
| self.assertEqual(generated_text, EXPECTED_GENERATED_TEXT) |
Oops, something went wrong.