Skip to content

Commit 7ea53ff

Browse files
authored
[https://nvbugs/5433545][fix] TestPhi4MiniInstruct::test_auto_dtype - Use max_seq_len=4096 to fallback to the short RoPE factor (#6895)
Signed-off-by: Michal Guzek <[email protected]>
1 parent 4b6cca0 commit 7ea53ff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integration/defs/accuracy/test_llm_api_pytorch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2334,7 +2334,7 @@ class TestPhi4MiniInstruct(LlmapiAccuracyTestHarness):
23342334
MODEL_PATH = f"{llm_models_root()}/Phi-4-mini-instruct"
23352335

23362336
def test_auto_dtype(self):
2337-
with LLM(self.MODEL_PATH) as llm:
2337+
with LLM(self.MODEL_PATH, max_seq_len=4096) as llm:
23382338
task = CnnDailymail(self.MODEL_NAME)
23392339
task.evaluate(llm)
23402340
task = MMLU(self.MODEL_NAME)

0 commit comments

Comments
 (0)