huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-09 19:34:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(python-client): stream not set

Browse Source
This commit is contained in:
OlivierDehaene 2023-03-08 16:46:50 +01:00
parent 5fd2dcb513
commit 1bc6742949
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clients/python/pyproject.toml
View File

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "text-generation" name = "text-generation"
version = "0.2.0" version = "0.2.1"
description = "Hugging Face Text Generation Python Client" description = "Hugging Face Text Generation Python Client"
license = "Apache-2.0" license = "Apache-2.0"
authors = ["Olivier Dehaene <olivier@huggingface.co>"] authors = ["Olivier Dehaene <olivier@huggingface.co>"]

2
clients/python/text_generation/__init__.py
View File

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
__version__ = "0.2.0" __version__ = "0.2.1"
from text_generation.client import Client, AsyncClient from text_generation.client import Client, AsyncClient
from text_generation.inference_api import InferenceAPIClient, InferenceAPIAsyncClient from text_generation.inference_api import InferenceAPIClient, InferenceAPIAsyncClient

2
clients/python/text_generation/client.py
View File

@ -191,7 +191,7 @@ class Client:
json=request.dict(), json=request.dict(),
headers=self.headers, headers=self.headers,
timeout=self.timeout, timeout=self.timeout,
stream=False, stream=True,
) )
if resp.status_code != 200: if resp.status_code != 200: