reorganize

formatting
pyannote · Jun 21, 2024 · 6b5f178 · 6b5f178
1 parent 5cd8fa0
commit 6b5f178
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pyannote/audio/core/pipeline.py b/pyannote/audio/core/pipeline.py
@@ -72,15 +72,14 @@ def from_pretrained(
             Path to model cache directory. Defaults to content of PYANNOTE_CACHE
             environment variable, or "~/.cache/torch/pyannote" when unset.
         """
+        if not cache_dir:
+            cache_dir = CACHE_DIR
 
         checkpoint_path = str(checkpoint_path)
 
         if os.path.isfile(checkpoint_path):
             config_yml = checkpoint_path
 
-        if not cache_dir:
-            cache_dir = CACHE_DIR
-
         else:
             if "@" in checkpoint_path:
                 model_id = checkpoint_path.split("@")[0]