fixed typo

2024-01-26 15:46:51 +01:00
parent c1ed0547b8
commit ff47d058c8
1 changed files with 25 additions and 25 deletions
@@ -226,7 +226,7 @@ class Diariser:
                warnings.warn(f"Model not found at {path_to_model}. "\
                    "Trying to find it nearby the config file.")
                
-                    pwd = file.split("/")[:-1]
+                pwd = model.split("/")[:-1]
                path_to_model = os.path.join(pwd, "pytorch_model.bin")

                if not os.path.exists(path_to_model):