Jan 2, 2024 · Jan 2, 2024 · Jan 2, 2024
diff --git a/pgml-extension/src/bindings/transformers/transformers.py b/pgml-extension/src/bindings/transformers/transformers.py
        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):
Original file line number	Diff line number	Diff line change
Expand Up		@@ -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):
Expand Down