Merge branch 'modelscope' of https://github.com/intel/neural-speed in…

…to modelscope
intel · Mar 6, 2024 · d693673 · d693673
2 parents 70cb61d + ce5a691
commit d693673
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)