mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-09-12 04:44:52 +00:00
Fixes.
This commit is contained in:
parent
1c58942133
commit
650ff012fe
@ -14,10 +14,6 @@ buildPythonPackage {
|
|||||||
|
|
||||||
build-system = [ poetry-core ];
|
build-system = [ poetry-core ];
|
||||||
|
|
||||||
nativeBuildInputs = [ ];
|
|
||||||
|
|
||||||
pythonRemoveDeps = [ ];
|
|
||||||
|
|
||||||
dependencies = [
|
dependencies = [
|
||||||
huggingface-hub
|
huggingface-hub
|
||||||
pydantic
|
pydantic
|
||||||
|
@ -590,7 +590,7 @@ async fn generate_stream_internal(
|
|||||||
let event = on_message_callback(stream_token);
|
let event = on_message_callback(stream_token);
|
||||||
yield Ok(event);
|
yield Ok(event);
|
||||||
}
|
}
|
||||||
// Yield event for lat token and compute timings
|
// Yield event for last token and compute timings
|
||||||
InferStreamResponse::End {
|
InferStreamResponse::End {
|
||||||
token,
|
token,
|
||||||
generated_text,
|
generated_text,
|
||||||
|
Loading…
Reference in New Issue
Block a user