reorganize

formatting
pyannote · Jun 15, 2024 · 7cf2d70 · 7cf2d70
1 parent 539985d
commit 7cf2d70
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]