From 63a75a79526c8cef008daeebf6b96d0fd1025083 Mon Sep 17 00:00:00 2001 From: yuanwu Date: Fri, 13 Jun 2025 05:56:40 +0000 Subject: [PATCH] Remove opt model Signed-off-by: yuanwu --- .../gaudi/server/integration-tests/test_model.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/backends/gaudi/server/integration-tests/test_model.py b/backends/gaudi/server/integration-tests/test_model.py index cb2bf6a9..95d0a275 100644 --- a/backends/gaudi/server/integration-tests/test_model.py +++ b/backends/gaudi/server/integration-tests/test_model.py @@ -165,20 +165,6 @@ TEST_CONFIGS = { "4", ], }, - "facebook/opt-125m": { - "model_id": "facebook/opt-125m", - "input": "What is Deep Learning?", - "expected_greedy_output": "\nAbout the Author\n\nAbout the Author\n\nAbout the Author\n\nAbout the Author\n\nAbout the Author\n\nAbout the Author\n\nAbout", - "expected_batch_output": "\nAbout the Author\n\nAbout the Author\n\nAbout the Author\n\nAbout the Author\n\nAbout the Author\n\nAbout the Author\n\nAbout", - "args": [ - "--max-input-tokens", - "512", - "--max-total-tokens", - "1024", - "--max-batch-size", - "4", - ], - }, "EleutherAI/gpt-j-6b": { "model_id": "EleutherAI/gpt-j-6b", "input": "What is Deep Learning?",