nomic-ai/gpt4all
1
1
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2024-09-20 01:27:27 +03: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
Show Stats 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') model = GPT4All(model_name='ggml-model-gpt4all-falcon-q4_0.bin')
prompt = 'hello' prompt = 'hello'
output = model.generate(prompt) output = model.generate(prompt)
assert isinstance(output, str)
assert len(output) > 0 assert len(output) > 0
@ -95,4 +95,5 @@ def test_inference_mpt():
model = GPT4All(model_name='ggml-mpt-7b-chat.bin') model = GPT4All(model_name='ggml-mpt-7b-chat.bin')
prompt = 'hello' prompt = 'hello'
output = model.generate(prompt) output = model.generate(prompt)
assert isinstance(output, str)
assert len(output) > 0 assert len(output) > 0