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)