diff --git a/neural_speed/__init__.py b/neural_speed/__init__.py index 4edd465af..65e71e937 100644 --- a/neural_speed/__init__.py +++ b/neural_speed/__init__.py @@ -91,7 +91,7 @@ def init(self, model_name, use_quant=True, use_gptq=False, use_awq=False, use_au if model_hub == "modelscope": from modelscope import AutoConfig self.config = AutoConfig.from_pretrained(model_name, trust_remote_code=True) - else: + else: self.config = AutoConfig.from_pretrained(model_name, trust_remote_code=True) model_type = Model.get_model_type(self.config) self.model_type = model_type diff --git a/neural_speed/convert/__init__.py b/neural_speed/convert/__init__.py index 025df9e2e..ed086badd 100644 --- a/neural_speed/convert/__init__.py +++ b/neural_speed/convert/__init__.py @@ -26,7 +26,7 @@ def convert_model(model, outfile, outtype="f32", model_hub="huggingface", use_qu if model_hub == "modelscope": from modelscope import AutoConfig config = AutoConfig.from_pretrained(model, trust_remote_code=True) - else: + else: config = AutoConfig.from_pretrained(model, trust_remote_code=True) model_type = model_maps.get(config.model_type, config.model_type)