Add handling for non-existent or empty model file paths in compute_model_hash() method

This commit is contained in:
Sergey Krashevich 2023-04-26 04:11:28 +03:00
parent 17e8a46c7d
commit bebb22dc11
No known key found for this signature in database
GPG Key ID: 625171324E7D3856

View File

@ -118,6 +118,10 @@ class ModelConfig(BaseModel):
}
def compute_model_hash(self) -> None:
if not self.path or not os.path.exists(self.path):
print("File does not exist or path is empty.")
self._model_hash = hashlib.md5(b"unknown").hexdigest()
else:
with open(self.path, "rb") as f:
file_hash = hashlib.md5()
while chunk := f.read(8192):