From 46e3c4b84a66efff60dd882173c54acc56e06356 Mon Sep 17 00:00:00 2001 From: Andrei Betlen Date: Mon, 1 May 2023 22:41:54 -0400 Subject: [PATCH] Fix --- tests/test_llama.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/test_llama.py b/tests/test_llama.py index 3ea19e0..b3426b8 100644 --- a/tests/test_llama.py +++ b/tests/test_llama.py @@ -151,9 +151,10 @@ def test_llama_server(): from fastapi.testclient import TestClient from llama_cpp.server.app import create_app, Settings - settings = Settings() - settings.model = MODEL - settings.vocab_only = True + settings = Settings( + model=MODEL, + vocab_only=True, + ) app = create_app(settings) client = TestClient(app) response = client.get("/v1/models")