mirror/OpenLLM
1
0
Fork 0
mirror of https://github.com/bentoml/OpenLLM.git synced 2026-05-05 06:12:43 -04:00
Code Issues Packages Projects Releases Wiki Activity

chore(ci): skip large models

Browse Source 
Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
aarnphm-ec2-dev
2023-07-16 05:52:42 +00:00
parent fc963c42ce
commit 5bb95652db
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models_test.py
View File

@@ -13,8 +13,11 @@
# limitations under the License.
from __future__ import annotations
import os
import typing as t
import pytest
if t.TYPE_CHECKING:
import openllm
@@ -32,6 +35,7 @@ def test_opt_implementation(prompt: str, llm: openllm.LLM[t.Any, t.Any]):
assert llm(prompt, temperature=0.9, top_k=8)
@pytest.mark.skipif(os.getenv("GITHUB_ACTIONS") is not None, "Model is too large for CI")
def test_baichuan_implementation(prompt: str, llm: openllm.LLM[t.Any, t.Any]):
assert llm(prompt)