From ead25a772a798fc8366abbd484645019ab1f59f7 Mon Sep 17 00:00:00 2001
From: SilasMarvin <19626586+SilasMarvin@users.noreply.github.com>
Date: Tue, 2 Jan 2024 13:34:22 -0800
Subject: [PATCH] Store errors in thread status
---
pgml-extension/src/bindings/transformers/transformers.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
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):
pFad - Phonifier reborn
Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.
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:
Alternative Proxy
pFad Proxy
pFad v3 Proxy
pFad v4 Proxy