From ca0a923d421905dcc37be47742419a95eb7dc331 Mon Sep 17 00:00:00 2001 From: eaidova Date: Wed, 4 Dec 2024 17:55:10 +0400 Subject: [PATCH] unblock wwb cli usage despite broken onnx --- tools/who_what_benchmark/whowhatbench/wwb.py | 14 +------------- 1 file changed, 1 insertion(+), 13 deletions(-) diff --git a/tools/who_what_benchmark/whowhatbench/wwb.py b/tools/who_what_benchmark/whowhatbench/wwb.py index f9aea15b47..058fafb61b 100644 --- a/tools/who_what_benchmark/whowhatbench/wwb.py +++ b/tools/who_what_benchmark/whowhatbench/wwb.py @@ -12,10 +12,7 @@ import pandas as pd from datasets import load_dataset from diffusers import DiffusionPipeline -from optimum.exporters.tasks import TasksManager -from optimum.intel import OVPipelineForText2Image -from optimum.intel.openvino import OVModelForCausalLM, OVModelForVisualCausalLM -from optimum.utils import NormalizedConfigManager, NormalizedTextConfig +from optimum.intel.openvino import OVModelForCausalLM, OVModelForVisualCausalLM, OVPipelineForText2Image from PIL import Image from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, AutoProcessor, AutoModel, AutoModelForVision2Seq import openvino as ov @@ -26,15 +23,6 @@ logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) -TasksManager._SUPPORTED_MODEL_TYPE["stablelm-epoch"] = ( - TasksManager._SUPPORTED_MODEL_TYPE["llama"] -) -NormalizedConfigManager._conf["stablelm-epoch"] = NormalizedTextConfig.with_args( - num_layers="num_hidden_layers", - num_attention_heads="num_attention_heads", -) - - class GenAIModelWrapper: """ A helper class to store additional attributes for GenAI models