Commit6d061ed

authored

Don't raise KeyError when popping in GGMLPipeline (#1464)

1 parent8823817 commit6d061edCopy full SHA for 6d061ed

File tree

-3

lines changed

-3

lines changed

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -211,9 +211,9 @@ class GGMLPipeline(object):`
`211`	`211`	`def__init__(self,model_name,**task):`
`212`	`212`	`importctransformers`
`213`	`213`
`214`		`-task.pop("model")`
`215`		`-task.pop("task")`
`216`		`-task.pop("device")`
	`214`	`+task.pop("model",None)`
	`215`	`+task.pop("task",None)`
	`216`	`+task.pop("device",None)`
`217`	`217`	`self.model=ctransformers.AutoModelForCausalLM.from_pretrained(`
`218`	`218`	`model_name,**task`
`219`	`219`	`)`

Comments

(0)