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
)
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