quality

huggingface · Jul 24, 2024 · a964c08 · a964c08
1 parent 33f28c6
commit a964c08
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/optimum/nvidia/utils/hub.py b/src/optimum/nvidia/utils/hub.py
@@ -108,7 +108,9 @@ def model_type_from_known_config(config: Dict[str, Any]) -> Optional[str]:
             ):  # Extracting (Llama)(ForCausalLM)
                 return match.group(1).lower()
             else:
-                raise RuntimeError(f"model_type {model_type} is not a valid model_type format")
+                raise RuntimeError(
+                    f"model_type {model_type} is not a valid model_type format"
+                )
         else:
             raise RuntimeError(f"Unable to process model_type: {model_type}")
     else:

diff --git a/tests/test_hub.py b/tests/test_hub.py
@@ -5,7 +5,6 @@
 import mock
 import pytest
 import torch.cuda
-from huggingface_hub import login
 from transformers import AutoConfig as HfAutoConfig
 from transformers import AutoModelForCausalLM as HfAutoModelForCausalLM