diff --git a/hf_dolly.py b/hf_dolly.py index fd091c0..70b04a7 100644 --- a/hf_dolly.py +++ b/hf_dolly.py @@ -17,7 +17,7 @@ from typing import List import torch from transformers import pipeline -from towhee.operator.base import PyOperator +from towhee.operator.base import PyOperator, SharedType class HuggingfaceDolly(PyOperator): @@ -60,3 +60,6 @@ class HuggingfaceDolly(PyOperator): model_list.sort() return model_list + @property + def shared_type(self): + return SharedType.Shareable