remove tmp if size overflow

pull/233/head
Stable Diffusion Daemon 2023-08-06 09:27:50 +09:00
parent 76128e7339
commit 2822f7cffa
1 changed files with 3 additions and 0 deletions

View File

@ -113,5 +113,8 @@ def dl(url, folder, filename, filepath):
os.rename(dl_file_path, file_path)
util.printD(f"File Downloaded to: {file_path}")
return file_path
elif downloaded_size > total_size:
# remove temporary-file
os.remove(dl_file_path)
return