From 519edd63fc8007eea59ddd952ef20d0837afc293 Mon Sep 17 00:00:00 2001 From: yzhao Date: Thu, 25 Aug 2022 20:17:16 +0800 Subject: [PATCH] fix bug --- modelscope/models/nlp/plug/plug_for_text_generation.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/modelscope/models/nlp/plug/plug_for_text_generation.py b/modelscope/models/nlp/plug/plug_for_text_generation.py index 8dceb6a8..fcd9081a 100644 --- a/modelscope/models/nlp/plug/plug_for_text_generation.py +++ b/modelscope/models/nlp/plug/plug_for_text_generation.py @@ -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)