Skip to content

Commit

Permalink
Add ChatGLM detokenization and tests
Browse files Browse the repository at this point in the history
  • Loading branch information
apaniukov committed Nov 17, 2023
1 parent 0f63c3d commit 0f1c1cc
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions modules/custom_operations/tests/run_tests.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,20 @@
# Copyright (C) 2018-2022 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
from pathlib import Path

from openvino.runtime import Core
from openvino.tools.mo import convert_model

import pytest
import numpy as np
import os
import sys


ext_path_dir = Path(os.getenv('CUSTOM_OP_LIB')).parent
if sys.platform == "win32":
# On Windows, with Python >= 3.8, DLLs are no longer imported from the PATH.
os.add_dll_directory(str(ext_path_dir.absolute()))


def run_test(ref_inputs, ref_res, test_onnx=False, threshold=1e-5):
Expand Down

0 comments on commit 0f1c1cc

Please sign in to comment.