enable accelerate

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
2 years ago · fc1cec2a53
1 changed files with 1 additions and 1 deletions
--- a/s_bert.py
+++ b/s_bert.py
@ -55,7 +55,7 @@ class ConvertModel(torch.nn.Module):
        return outs['sentence_embedding']


-# @accelerate
+@accelerate
 class Model:
    def __init__(self, model_name, device):
        self.device = device