note: Time of check to time of use

This commit is contained in:
teleprint-me 2024-05-07 21:56:07 -04:00
parent cbfed5b01a
commit f7dda38b10
No known key found for this signature in database
GPG key ID: B0D11345E65C4D48

View file

@ -109,7 +109,7 @@ for model in models:
model_config_path = f"{model_name_or_path}/config.json"
# check dir path
if not os.path.exists(model_name_or_path):
if not os.path.exists(model_name_or_path): # TOCTOU
os.makedirs(model_name_or_path)
else:
logger.info(f"Directory {model_name_or_path} already exists - skipping")