From 8b6356527e19e8797ea594508c26ec13afc206c7 Mon Sep 17 00:00:00 2001 From: Kevin Zimmerman <4733573+kczimm@users.noreply.github.com> Date: Tue, 14 May 2024 11:59:37 -0500 Subject: [PATCH] pop gently in GGMLPipeline --- pgml-extension/src/bindings/transformers/transformers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pgml-extension/src/bindings/transformers/transformers.py b/pgml-extension/src/bindings/transformers/transformers.py index b0d456104..782dd7908 100644 --- a/pgml-extension/src/bindings/transformers/transformers.py +++ b/pgml-extension/src/bindings/transformers/transformers.py @@ -211,9 +211,9 @@ class GGMLPipeline(object): def __init__(self, model_name, **task): import ctransformers - task.pop("model") - task.pop("task") - task.pop("device") + task.pop("model", None) + task.pop("task", None) + task.pop("device", None) self.model = ctransformers.AutoModelForCausalLM.from_pretrained( model_name, **task )