fixed path

This commit is contained in:
Jaikinator
2024-01-26 15:56:10 +01:00
parent ff47d058c8
commit b45a601119
+2
View File
@@ -227,6 +227,8 @@ class Diariser:
"Trying to find it nearby the config file.") "Trying to find it nearby the config file.")
pwd = model.split("/")[:-1] pwd = model.split("/")[:-1]
pwd = "/".join(pwd)
path_to_model = os.path.join(pwd, "pytorch_model.bin") path_to_model = os.path.join(pwd, "pytorch_model.bin")
if not os.path.exists(path_to_model): if not os.path.exists(path_to_model):