github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-09-08 03:49:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

bindings/python: type assert

Browse Source
This commit is contained in:
Aaron Miller
2023-06-30 16:17:46 -07:00
committed by AT
parent 57dc0c8953
commit 3599663a22
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gpt4all-bindings/python/gpt4all/tests/test_gpt4all.py
View File

@@ -87,7 +87,7 @@ def test_inference_falcon():
model = GPT4All(model_name='ggml-model-gpt4all-falcon-q4_0.bin')
prompt = 'hello'
output = model.generate(prompt)
assert isinstance(output, str)
assert len(output) > 0
@@ -95,4 +95,5 @@ def test_inference_mpt():
model = GPT4All(model_name='ggml-mpt-7b-chat.bin')
prompt = 'hello'
output = model.generate(prompt)
assert isinstance(output, str)
assert len(output) > 0