Improve benchmark (#1140)

This commit is contained in:
Lianmin Zheng
2024-08-17 17:43:23 -07:00
committed by GitHub
parent cdc8d60752
commit 57d0bd91ec
8 changed files with 111 additions and 678 deletions

View File

@@ -113,30 +113,7 @@ def call_generate_srt_raw(prompt, temperature, max_tokens, stop=None, url=None):
def call_generate_ginfer(prompt, temperature, max_tokens, stop=None, url=None):
import grpc
from ginfer import sampler_pb2, sampler_pb2_grpc
sampler_channel = grpc.insecure_channel(url.replace("http://", ""))
sampler = sampler_pb2_grpc.SamplerStub(sampler_channel)
if stop is None:
stop_strings = None
else:
stop_strings = [stop]
sample_request = sampler_pb2.SampleTextRequest(
prompt=prompt,
settings=sampler_pb2.SampleSettings(
max_len=max_tokens,
rng_seed=0,
temperature=max(temperature, 1e-7),
nucleus_p=1,
stop_strings=stop_strings,
),
)
stream = sampler.SampleText(sample_request)
response = "".join([x.text for x in stream])
return response
raise NotImplementedError()
def call_generate_guidance(