Update expected values (after switching to A10) - part 5 (#39205)
* fix * fix * fix * fix * fix * fix * fix * fix * fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -404,10 +404,15 @@ class Emu3IntegrationTest(unittest.TestCase):
|
|||||||
"USER: 64*64Describe what do you see here? ASSISTANT: The image depicts a black panther in a crouched position. The panther's body is elongated and its head is lowered, suggesting a state of alertness or readiness. The animal's",
|
"USER: 64*64Describe what do you see here? ASSISTANT: The image depicts a black panther in a crouched position. The panther's body is elongated and its head is lowered, suggesting a state of alertness or readiness. The animal's",
|
||||||
"USER: 64*64What can you say about the image? ASSISTANT: The image depicts a serene natural landscape. The foreground consists of a grassy area with some patches of bare earth. The middle ground shows a gently sloping hill with a reddish-brown hue,",
|
"USER: 64*64What can you say about the image? ASSISTANT: The image depicts a serene natural landscape. The foreground consists of a grassy area with some patches of bare earth. The middle ground shows a gently sloping hill with a reddish-brown hue,",
|
||||||
],
|
],
|
||||||
("cuda", 7): [
|
(None, None): [
|
||||||
"USER: 64*64Describe what do you see here? ASSISTANT: The image depicts a black panther in a crouched position. The panther's body is elongated and curved, with its head lowered and ears pointed forward, suggesting alertness or focus.",
|
"USER: 64*64Describe what do you see here? ASSISTANT: The image depicts a black panther in a crouched position. The panther's body is elongated and curved, with its head lowered and ears pointed forward, suggesting alertness or focus.",
|
||||||
"USER: 64*64What can you say about the image? ASSISTANT: The image depicts a serene natural landscape. The foreground consists of a grassy area with some patches of bare earth. The middle ground shows a steep, reddish-brown cliff, which could be a",
|
"USER: 64*64What can you say about the image? ASSISTANT: The image depicts a serene natural landscape. The foreground consists of a grassy area with some patches of bare earth. The middle ground shows a steep, reddish-brown cliff, which could be a",
|
||||||
],
|
],
|
||||||
|
# We switch to A10 on 2025/06/29, and A10 gives strange values
|
||||||
|
("cuda", 8): [
|
||||||
|
'USER: 64*64Describe what do you see here? ASSISTANT: 1.Filed with 1.Computing theComputing.Computing.',
|
||||||
|
'USER: 64*64What can you say about the image? ASSISTANT: 1.Filed with theComputing theComputing.Computing.',
|
||||||
|
],
|
||||||
}
|
}
|
||||||
) # fmt: skip
|
) # fmt: skip
|
||||||
EXPECTED_TEXT_COMPLETION = EXPECTED_TEXT_COMPLETIONS.get_expectation()
|
EXPECTED_TEXT_COMPLETION = EXPECTED_TEXT_COMPLETIONS.get_expectation()
|
||||||
@@ -433,7 +438,9 @@ class Emu3IntegrationTest(unittest.TestCase):
|
|||||||
EXPECTED_TEXT_COMPLETIONS = Expectations(
|
EXPECTED_TEXT_COMPLETIONS = Expectations(
|
||||||
{
|
{
|
||||||
("xpu", 3): ['USER: 64*6464*64What do these two images have in common? ASSISTANT: The two images both depict a rhinoceros, yet they are significantly different in terms of focus and clarity. The rhinoceros in the upper image is in sharp focus, showing detailed textures'],
|
("xpu", 3): ['USER: 64*6464*64What do these two images have in common? ASSISTANT: The two images both depict a rhinoceros, yet they are significantly different in terms of focus and clarity. The rhinoceros in the upper image is in sharp focus, showing detailed textures'],
|
||||||
("cuda", 7): ["USER: 64*6464*64What do these two images have in common? ASSISTANT: Both images feature a black animal, but they are not the same animal. The top image shows a close-up of a black cow's head, while the bottom image depicts a black cow in a natural"],
|
(None, None): ["USER: 64*6464*64What do these two images have in common? ASSISTANT: Both images feature a black animal, but they are not the same animal. The top image shows a close-up of a black cow's head, while the bottom image depicts a black cow in a natural"],
|
||||||
|
# We switch to A10 on 2025/06/29, and A10 gives strange values
|
||||||
|
("cuda", 8): ['USER: 64*6464*64What do these two images have in common? ASSISTANT:Computing.Filed.Filed.11.Computing theComputing.Computing.'],
|
||||||
}
|
}
|
||||||
) # fmt: skip
|
) # fmt: skip
|
||||||
EXPECTED_TEXT_COMPLETION = EXPECTED_TEXT_COMPLETIONS.get_expectation()
|
EXPECTED_TEXT_COMPLETION = EXPECTED_TEXT_COMPLETIONS.get_expectation()
|
||||||
|
|||||||
@@ -24,6 +24,7 @@ from datasets import Audio, load_dataset
|
|||||||
|
|
||||||
from transformers import AutoProcessor, EncodecConfig
|
from transformers import AutoProcessor, EncodecConfig
|
||||||
from transformers.testing_utils import (
|
from transformers.testing_utils import (
|
||||||
|
Expectations,
|
||||||
is_torch_available,
|
is_torch_available,
|
||||||
require_torch,
|
require_torch,
|
||||||
slow,
|
slow,
|
||||||
@@ -459,10 +460,21 @@ class EncodecIntegrationTest(unittest.TestCase):
|
|||||||
"1.5": 0.0025,
|
"1.5": 0.0025,
|
||||||
"24.0": 0.0015,
|
"24.0": 0.0015,
|
||||||
}
|
}
|
||||||
expected_codesums = {
|
|
||||||
"1.5": [371955],
|
expectations = Expectations(
|
||||||
"24.0": [6659962],
|
{
|
||||||
}
|
(None, None): {
|
||||||
|
"1.5": [371955],
|
||||||
|
"24.0": [6659962],
|
||||||
|
},
|
||||||
|
("cuda", 8): {
|
||||||
|
"1.5": [371955],
|
||||||
|
"24.0": [6655079],
|
||||||
|
},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
expected_codesums = expectations.get_expectation()
|
||||||
|
|
||||||
librispeech_dummy = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
|
librispeech_dummy = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
|
||||||
model_id = "facebook/encodec_24khz"
|
model_id = "facebook/encodec_24khz"
|
||||||
|
|
||||||
@@ -513,10 +525,21 @@ class EncodecIntegrationTest(unittest.TestCase):
|
|||||||
"3.0": 0.001,
|
"3.0": 0.001,
|
||||||
"24.0": 0.0005,
|
"24.0": 0.0005,
|
||||||
}
|
}
|
||||||
expected_codesums = {
|
|
||||||
"3.0": [144259, 146765, 156435, 176871, 161971],
|
expectations = Expectations(
|
||||||
"24.0": [1568553, 1294948, 1306190, 1464747, 1663150],
|
{
|
||||||
}
|
(None, None): {
|
||||||
|
"3.0": [144259, 146765, 156435, 176871, 161971],
|
||||||
|
"24.0": [1568553, 1294948, 1306190, 1464747, 1663150],
|
||||||
|
},
|
||||||
|
("cuda", 8): {
|
||||||
|
"3.0": [144259, 146765, 156205, 176871, 161971],
|
||||||
|
"24.0": [1566878, 1300459, 1310165, 1464747, 1663150],
|
||||||
|
},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
expected_codesums = expectations.get_expectation()
|
||||||
|
|
||||||
librispeech_dummy = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
|
librispeech_dummy = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
|
||||||
model_id = "facebook/encodec_48khz"
|
model_id = "facebook/encodec_48khz"
|
||||||
|
|
||||||
|
|||||||
@@ -21,6 +21,8 @@ import pytest
|
|||||||
|
|
||||||
from transformers import FalconH1Config, is_torch_available
|
from transformers import FalconH1Config, is_torch_available
|
||||||
from transformers.testing_utils import (
|
from transformers.testing_utils import (
|
||||||
|
Expectations,
|
||||||
|
get_device_properties,
|
||||||
require_torch,
|
require_torch,
|
||||||
require_torch_gpu,
|
require_torch_gpu,
|
||||||
slow,
|
slow,
|
||||||
@@ -484,7 +486,7 @@ class FalconH1ModelIntegrationTest(unittest.TestCase):
|
|||||||
"""
|
"""
|
||||||
An integration test for Falcon-H1.
|
An integration test for Falcon-H1.
|
||||||
"""
|
"""
|
||||||
EXPECTED_TEXT = """
|
EXPECTED_TEXT_DEFAULT = """
|
||||||
user
|
user
|
||||||
Tell me about the french revolution.
|
Tell me about the french revolution.
|
||||||
assistant
|
assistant
|
||||||
@@ -503,9 +505,43 @@ class FalconH1ModelIntegrationTest(unittest.TestCase):
|
|||||||
4. **Rise of the Jacobins and Reign of Terror (1793–1794)**: Radical leaders like Maximilien Robespierre sought to purge France of counter-revolutionaries, leading to mass executions and widespread fear.
|
4. **Rise of the Jacobins and Reign of Terror (1793–1794)**: Radical leaders like Maximilien Robespierre sought to purge France of counter-revolutionaries, leading to mass executions and widespread fear.
|
||||||
5. **Thermidorian Reaction
|
5. **Thermidorian Reaction
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
EXPECTED_TEXT_A10 = """
|
||||||
|
user
|
||||||
|
Tell me about the french revolution.
|
||||||
|
assistant
|
||||||
|
The French Revolution (1789–1799) was a period of profound social upheaval and radical political change in France that fundamentally transformed the nation and had far-reaching effects on the rest of Europe and the world. Here are the key aspects of the revolution:
|
||||||
|
|
||||||
|
### **Causes**
|
||||||
|
1. **Economic Crisis**: France was in severe financial trouble due to costly wars (particularly the American Revolution), extravagant spending by the monarchy, and an inefficient tax system.
|
||||||
|
2. **Social Inequality**: The privileged classes (the nobility and clergy) enjoyed immense wealth and power, while the majority of the population (the Third Estate, comprising commoners) faced poverty and lack of representation.
|
||||||
|
3. **Enlightenment Ideas**: Philosophers like Voltaire, Rousseau, and Montesquieu inspired ideas of liberty, equality, and popular sovereignty, which fueled revolutionary fervor.
|
||||||
|
4. **Political Instability**: The absolute monarchy under King Louis XVI proved unable to address the nation's problems, leading to growing discontent.
|
||||||
|
|
||||||
|
### **Key Events**
|
||||||
|
1. **Estates-General (1789)**: The Third Estate broke away and formed the National Assembly, forcing King Louis XVI to convene the Estates-General, an old legislative body, to address the financial crisis.
|
||||||
|
2. **Storming of the Bastille (July 14, 1789)**: A symbol of royal tyranny, the Bastille fortress was stormed by revolutionaries, sparking widespread rebellion.
|
||||||
|
3. **Declaration of the Rights of Man and of the Citizen (August 1789)**: This foundational document proclaimed liberty, equality, and fraternity as fundamental rights.
|
||||||
|
4. **Abolition of Feudalism (November 1789)**: The National Assembly abolished feudal privileges, redistributing church lands to the people.
|
||||||
|
5. **Tennis Court Oath (May 5, 1789)**: The National Assembly members, meeting on a tennis court, pledged to continue their work until a new constitution was established.
|
||||||
|
6.
|
||||||
|
"""
|
||||||
|
|
||||||
|
expected_texts = Expectations(
|
||||||
|
{
|
||||||
|
(None, None): EXPECTED_TEXT_DEFAULT,
|
||||||
|
("cuda", 8): EXPECTED_TEXT_A10,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
EXPECTED_TEXT = expected_texts.get_expectation()
|
||||||
# Remove the first char (`\n`) and the consecutive whitespaces caused by the formatting.
|
# Remove the first char (`\n`) and the consecutive whitespaces caused by the formatting.
|
||||||
EXPECTED_TEXT = EXPECTED_TEXT.strip().replace(" " * 12, "")
|
EXPECTED_TEXT = EXPECTED_TEXT.strip().replace(" " * 12, "")
|
||||||
|
|
||||||
|
device_properties = get_device_properties()
|
||||||
|
# For A10, there is an ending " "
|
||||||
|
if device_properties[0] == "cuda" and device_properties[1] == 8:
|
||||||
|
EXPECTED_TEXT = EXPECTED_TEXT + " "
|
||||||
|
|
||||||
model_id = "tiiuae/Falcon-H1-1.5B-Deep-Instruct"
|
model_id = "tiiuae/Falcon-H1-1.5B-Deep-Instruct"
|
||||||
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
||||||
model = FalconH1ForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
|
model = FalconH1ForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
|
||||||
|
|||||||
@@ -403,8 +403,7 @@ class Gemma3IntegrationTest(unittest.TestCase):
|
|||||||
EXPECTED_TEXTS = Expectations(
|
EXPECTED_TEXTS = Expectations(
|
||||||
{
|
{
|
||||||
("xpu", 3): ['user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach with turquoise water in the background. It looks like a lovely,'],
|
("xpu", 3): ['user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach with turquoise water in the background. It looks like a lovely,'],
|
||||||
("cuda", 7): ['user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach with turquoise water in the background. It looks like a lovely,'],
|
("cuda", 8): ['user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach with turquoise water and a distant coastline in the background. It looks'],
|
||||||
("cuda", 8): ['user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach next to a turquoise ocean. It looks like a very sunny and'],
|
|
||||||
("rocm", (9, 5)): ['user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach with turquoise water and a distant coastline in the background. It looks'],
|
("rocm", (9, 5)): ['user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach with turquoise water and a distant coastline in the background. It looks'],
|
||||||
}
|
}
|
||||||
) # fmt: skip
|
) # fmt: skip
|
||||||
@@ -452,11 +451,10 @@ class Gemma3IntegrationTest(unittest.TestCase):
|
|||||||
'user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach next to a turquoise ocean. It looks like a very sunny and',
|
'user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach next to a turquoise ocean. It looks like a very sunny and',
|
||||||
'user\nYou are a helpful assistant.\n\n\n\n\n\n\n\n\n\nAre these images identical?\nmodel\nNo, these images are not identical. They depict very different scenes:\n\n* **Image 1** shows a cow standing on a beach.',
|
'user\nYou are a helpful assistant.\n\n\n\n\n\n\n\n\n\nAre these images identical?\nmodel\nNo, these images are not identical. They depict very different scenes:\n\n* **Image 1** shows a cow standing on a beach.',
|
||||||
],
|
],
|
||||||
("cuda", 7): [],
|
|
||||||
("cuda", 8):
|
("cuda", 8):
|
||||||
[
|
[
|
||||||
'user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach next to a turquoise ocean. It looks like a very sunny and',
|
'user\nYou are a helpful assistant.\n\n\n\n\n\nWhat is shown in this image?\nmodel\nCertainly! \n\nThe image shows a brown and white cow standing on a sandy beach with turquoise water and a distant island in the background. It looks',
|
||||||
'user\nYou are a helpful assistant.\n\n\n\n\n\n\n\n\n\nAre these images identical?\nmodel\nNo, these images are not identical. They depict very different scenes:\n\n* **Image 1** shows a cow standing on a beach.',
|
'user\nYou are a helpful assistant.\n\n\n\n\n\n\n\n\n\nAre these images identical?\nmodel\nNo, these images are not identical. They depict very different scenes. \n\n* **Image 1** shows a cow standing on a beach'
|
||||||
],
|
],
|
||||||
("rocm", (9, 5)):
|
("rocm", (9, 5)):
|
||||||
[
|
[
|
||||||
|
|||||||
@@ -120,10 +120,14 @@ class GlmIntegrationTest(unittest.TestCase):
|
|||||||
|
|
||||||
def test_model_9b_eager(self):
|
def test_model_9b_eager(self):
|
||||||
expected_texts = Expectations({
|
expected_texts = Expectations({
|
||||||
("cuda", None): [
|
(None, None): [
|
||||||
"Hello I am doing a project on the history of the internetSolution:\n\nStep 1: Introduction\nThe history of the",
|
"Hello I am doing a project on the history of the internetSolution:\n\nStep 1: Introduction\nThe history of the",
|
||||||
"Hi today I am going to show you how to make a simple and easy to make a DIY paper flower.",
|
"Hi today I am going to show you how to make a simple and easy to make a DIY paper flower.",
|
||||||
],
|
],
|
||||||
|
("cuda", 8): [
|
||||||
|
'Hello I am doing a project on the history of the internetSolution:\n\nStep 1: Introduction\nThe history of the',
|
||||||
|
'Hi today I am going to show you how to make a simple and easy to make a DIY paper lantern.',
|
||||||
|
],
|
||||||
("rocm", (9, 5)) : [
|
("rocm", (9, 5)) : [
|
||||||
"Hello I am doing a project on the history of the internetSolution:\n\nStep 1: Introduction\nThe history of the",
|
"Hello I am doing a project on the history of the internetSolution:\n\nStep 1: Introduction\nThe history of the",
|
||||||
"Hi today I am going to show you how to make a simple and easy to make a paper airplane. First",
|
"Hi today I am going to show you how to make a simple and easy to make a paper airplane. First",
|
||||||
|
|||||||
@@ -87,7 +87,8 @@ class HeliumIntegrationTest(unittest.TestCase):
|
|||||||
expected_texts = Expectations(
|
expected_texts = Expectations(
|
||||||
{
|
{
|
||||||
("rocm", (9, 5)): ["Hello, today is a great day to start a new project. I have been working on a new project for a while now, and I"],
|
("rocm", (9, 5)): ["Hello, today is a great day to start a new project. I have been working on a new project for a while now, and I"],
|
||||||
("cuda", None): ["Hello, today is a great day to start a new project. I have been working on a new project for a while now and I have"],
|
(None, None): ["Hello, today is a great day to start a new project. I have been working on a new project for a while now and I have"],
|
||||||
|
("cuda", 8): ['Hello, today is a great day to start a new project. I have been working on a new project for a while now, and I'],
|
||||||
}
|
}
|
||||||
) # fmt: skip
|
) # fmt: skip
|
||||||
EXPECTED_TEXTS = expected_texts.get_expectation()
|
EXPECTED_TEXTS = expected_texts.get_expectation()
|
||||||
|
|||||||
@@ -545,7 +545,8 @@ class PaliGemmaForConditionalGenerationIntegrationTest(unittest.TestCase):
|
|||||||
expected_decoded_texts = Expectations(
|
expected_decoded_texts = Expectations(
|
||||||
{
|
{
|
||||||
("rocm", (9, 5)): "detect shoe\n<loc0051><loc0309><loc0708><loc0644> shoe",
|
("rocm", (9, 5)): "detect shoe\n<loc0051><loc0309><loc0708><loc0644> shoe",
|
||||||
("cuda", None): "detect shoe\n<loc0051><loc0309><loc0708><loc0646> shoe",
|
(None, None): "detect shoe\n<loc0051><loc0309><loc0708><loc0646> shoe",
|
||||||
|
("cuda", 8): "detect shoe\n<loc0045><loc0309><loc0708><loc0646> shoe",
|
||||||
}
|
}
|
||||||
) # fmt: skip
|
) # fmt: skip
|
||||||
EXPECTED_DECODED_TEXT = expected_decoded_texts.get_expectation()
|
EXPECTED_DECODED_TEXT = expected_decoded_texts.get_expectation()
|
||||||
|
|||||||
@@ -256,6 +256,9 @@ class Qwen2IntegrationTest(unittest.TestCase):
|
|||||||
("cuda", None): [
|
("cuda", None): [
|
||||||
"My favourite condiment is 100% natural, organic, gluten free, vegan, and free from preservatives. I"
|
"My favourite condiment is 100% natural, organic, gluten free, vegan, and free from preservatives. I"
|
||||||
],
|
],
|
||||||
|
("cuda", 8): [
|
||||||
|
"My favourite condiment is 100% natural, organic, gluten free, vegan, and vegetarian. I love to use"
|
||||||
|
],
|
||||||
("rocm", (9, 5)): [
|
("rocm", (9, 5)): [
|
||||||
"My favourite condiment is 100% natural, organic, gluten free, vegan, and vegetarian. I love to use"
|
"My favourite condiment is 100% natural, organic, gluten free, vegan, and vegetarian. I love to use"
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -466,6 +466,10 @@ class Qwen2VLIntegrationTest(unittest.TestCase):
|
|||||||
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and intelligent nature, making them popular choices',
|
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and intelligent nature, making them popular choices',
|
||||||
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and intelligent nature, making them popular pets',
|
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and intelligent nature, making them popular pets',
|
||||||
],
|
],
|
||||||
|
("cuda", 8): [
|
||||||
|
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and intelligent nature, making them popular choices',
|
||||||
|
'system\nYou are a helpful assistant.\nuser\nWhat kind of dog is this?\nassistant\nThe dog in the picture appears to be a Labrador Retriever. Labradors are known for their friendly and intelligent nature, making them popular choices'
|
||||||
|
],
|
||||||
}
|
}
|
||||||
) # fmt: skip
|
) # fmt: skip
|
||||||
EXPECTED_DECODED_TEXT = EXPECTED_DECODED_TEXTS.get_expectation()
|
EXPECTED_DECODED_TEXT = EXPECTED_DECODED_TEXTS.get_expectation()
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ import pytest
|
|||||||
|
|
||||||
from transformers import Starcoder2Config, is_torch_available
|
from transformers import Starcoder2Config, is_torch_available
|
||||||
from transformers.testing_utils import (
|
from transformers.testing_utils import (
|
||||||
|
Expectations,
|
||||||
require_bitsandbytes,
|
require_bitsandbytes,
|
||||||
require_flash_attn,
|
require_flash_attn,
|
||||||
require_torch,
|
require_torch,
|
||||||
@@ -148,10 +149,20 @@ class Starcoder2IntegrationTest(unittest.TestCase):
|
|||||||
|
|
||||||
@require_bitsandbytes
|
@require_bitsandbytes
|
||||||
def test_starcoder2_batched_generation_4bit(self):
|
def test_starcoder2_batched_generation_4bit(self):
|
||||||
EXPECTED_TEXT = [
|
expectations = Expectations(
|
||||||
'Hello my name is Younes and I am a student at the University of Maryland. I am currently working on a project that is related to the topic of "How to make a game". I am currently working on a project',
|
{
|
||||||
'def hello_world():\n\treturn "Hello World"\n\n@app.route(\'/hello/<name>\')\ndef hello_name(name):\n\treturn "Hello " + name\n\n@app.route',
|
(None, None): [
|
||||||
]
|
'Hello my name is Younes and I am a student at the University of Maryland. I am currently working on a project that is related to the topic of "How to make a game". I am currently working on a project',
|
||||||
|
'def hello_world():\n\treturn "Hello World"\n\n@app.route(\'/hello/<name>\')\ndef hello_name(name):\n\treturn "Hello " + name\n\n@app.route',
|
||||||
|
],
|
||||||
|
("cuda", 8): [
|
||||||
|
"Hello my name is Younes and I am a student at the University of Maryland. I am currently working on a project that is aimed at creating a new way of learning. I am hoping to create a new way of",
|
||||||
|
'def hello_world():\n\treturn "Hello World"\n\n@app.route(\'/hello/<name>\')\ndef hello_name(name):\n\treturn "Hello " + name\n\n@app.route',
|
||||||
|
],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
EXPECTED_TEXT = expectations.get_expectation()
|
||||||
|
|
||||||
model_id = "bigcode/starcoder2-7b"
|
model_id = "bigcode/starcoder2-7b"
|
||||||
|
|
||||||
model = Starcoder2ForCausalLM.from_pretrained(model_id, load_in_4bit=True)
|
model = Starcoder2ForCausalLM.from_pretrained(model_id, load_in_4bit=True)
|
||||||
|
|||||||
Reference in New Issue
Block a user