azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 23:16:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

safe model saving

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2025-05-31 19:09:42 +03:00
parent c9f2c167e2
commit b59392b78b
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api_client.py
+3
View File
@@ -1,5 +1,6 @@
import io import io
import json import json
import os
from http import HTTPStatus from http import HTTPStatus
from os import path from os import path
@@ -120,6 +121,8 @@ class ApiClient:
part_big = resource_encrypted[part_small_size:] part_big = resource_encrypted[part_small_size:]
self.cdn_manager.upload(constants.MODELS_FOLDER, big_part_name, part_big) self.cdn_manager.upload(constants.MODELS_FOLDER, big_part_name, part_big)
os.makedirs(folder, exist_ok=True)
with open(path.join(folder, big_part_name), 'wb') as f: with open(path.join(folder, big_part_name), 'wb') as f:
f.write(part_big) f.write(part_big)
self.upload_file(small_part_name, part_small, constants.MODELS_FOLDER) self.upload_file(small_part_name, part_small, constants.MODELS_FOLDER)