File tree Expand file tree Collapse file tree 3 files changed +4
-4
lines changed
Expand file tree Collapse file tree 3 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -24,7 +24,7 @@ def test_rerank():
2424 })
2525 assert res .status_code == 200
2626 assert len (res .body ["results" ]) == 4
27-
27+
2828 most_relevant = res .body ["results" ][0 ]
2929 least_relevant = res .body ["results" ][0 ]
3030 for doc in res .body ["results" ]:
Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ def test_slot_save_restore():
4747 })
4848 assert res .status_code == 200
4949 assert res .body ["n_restored" ] == 84
50-
50+
5151 # Since we have cache, slot 0 should only process the last tokens
5252 res = server .make_request ("POST" , "/completion" , data = {
5353 "prompt" : "What is the capital of Germany?" ,
Original file line number Diff line number Diff line change @@ -292,7 +292,7 @@ def tinyllama_infill() -> ServerProcess:
292292 server .temperature = 0.0
293293 server .seed = 42
294294 return server
295-
295+
296296 @staticmethod
297297 def stories15m_moe () -> ServerProcess :
298298 server = ServerProcess ()
@@ -306,7 +306,7 @@ def stories15m_moe() -> ServerProcess:
306306 server .temperature = 0.0
307307 server .seed = 42
308308 return server
309-
309+
310310 @staticmethod
311311 def jina_reranker_tiny () -> ServerProcess :
312312 server = ServerProcess ()
You can’t perform that action at this time.
0 commit comments