(fix) only decode chunk when it's not a str

This commit is contained in:
ishaan-jaff 2023-11-16 12:23:49 -08:00
parent 95642df0a4
commit 55a054f3f6

View file

@ -4367,6 +4367,7 @@ class CustomStreamWrapper:
def handle_huggingface_chunk(self, chunk):
try:
if type(chunk) != str:
chunk = chunk.decode("utf-8") # DO NOT REMOVE this: This is required for HF inference API + Streaming
text = ""
is_finished = False