From 2d28da2f68d0f282faef8edf8e41f2c406e90cc9 Mon Sep 17 00:00:00 2001
From: "junjie.jiang" <junjie.jiang@zilliz.com>
Date: Thu, 1 Jun 2023 10:21:08 +0800
Subject: [PATCH] Add sharable

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
---
 hf_dolly.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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