fixed typo
This commit is contained in:
@@ -226,7 +226,7 @@ class Diariser:
|
|||||||
warnings.warn(f"Model not found at {path_to_model}. "\
|
warnings.warn(f"Model not found at {path_to_model}. "\
|
||||||
"Trying to find it nearby the config file.")
|
"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")
|
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):
|
||||||
|
|||||||
Reference in New Issue
Block a user