Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
yzhao
2022-08-25 20:17:16 +08:00
parent 78ab0cf16d
commit 519edd63fc
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modelscope/models/nlp/plug/plug_for_text_generation.py
View File

@@ -32,8 +32,7 @@ class PlugForTextGeneration(DistributedTorchModel):
return res
def _instantiate_one(self, rank, model_dir, **kwargs):
cfg = read_config(model_dir)
self.__class__.model = DistributedPlug(model_dir, rank, **cfg.model, **kwargs)
self.__class__.model = DistributedPlug(model_dir, rank, **kwargs)