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

feat: 1.2 APIs (#821)

Browse Source 
Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Aaron Pham
2024-03-15 03:49:19 -04:00
committed by GitHub
parent e3392476be
commit 072b3e97ec
116 changed files with 4451 additions and 6144 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
openllm-python/tests/package_test.py
View File

@@ -40,7 +40,7 @@ def test_general_build_from_local(tmp_path_factory: pytest.TempPathFactory):
llm = openllm.LLM(model_id=HF_INTERNAL_T5_TESTING, serialisation='legacy')
llm.model.save_pretrained(str(local_path))
llm.tokenizer.save_pretrained(str(local_path))
llm._tokenizer.save_pretrained(str(local_path))
assert openllm.build('flan-t5', model_id=local_path.resolve().__fspath__(), model_version='local')