Browse Source

[to #42322933] Fix bug for distributed gpt3

修复 Pipeline 更新后 DistributedGPT3Pipeline 出现的报错
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10614071
master
hemu.zp yingda.chen 3 years ago
parent
commit
dc1b88b396
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      modelscope/pipelines/base.py

+ 1
- 0
modelscope/pipelines/base.py View File

@@ -366,6 +366,7 @@ class DistributedPipeline(Pipeline):
master_port=master_port,
**self.cfg.model,
**kwargs), ranks)
self.models = []

def __del__(self):
if hasattr(self, 'model_pool') and self.model_pool is not None:


Loading…
Cancel
Save