Rename servers to engines (#152)
This commit is contained in:
@@ -84,7 +84,7 @@ def run_cacheflow(
|
||||
seed=seed,
|
||||
)
|
||||
|
||||
# Add the requests to the server.
|
||||
# Add the requests to the engine.
|
||||
for prompt, _, output_len in requests:
|
||||
sampling_params = SamplingParams(
|
||||
n=n,
|
||||
@@ -103,7 +103,7 @@ def run_cacheflow(
|
||||
|
||||
start = time.time()
|
||||
# FIXME(woosuk): Do use internal method.
|
||||
llm._run_server(use_tqdm=True)
|
||||
llm._run_engine(use_tqdm=True)
|
||||
end = time.time()
|
||||
return end - start
|
||||
|
||||
|
||||
Reference in New Issue
Block a user