diff --git a/pgml-extension/src/bindings/transformers/transformers.py b/pgml-extension/src/bindings/transformers/transformers.py index 83608ed48..83d953fdd 100644 --- a/pgml-extension/src/bindings/transformers/transformers.py +++ b/pgml-extension/src/bindings/transformers/transformers.py @@ -187,10 +187,9 @@ def streaming_worker(worker_threads, model, **kwargs): worker_threads.update_thread(thread_id, "Error setting data") try: model.generate(**kwargs) - except BaseException as error: - print(f"Error in streaming_worker: {error}", file=sys.stderr) - finally: worker_threads.delete_thread(thread_id) + except BaseException as error: + worker_threads.update_thread(thread_id, f"Error in streaming_worker: {error}") class GGMLPipeline(object):
Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.
Alternative Proxies: