[Feature] utils : weightフォルダ名をweightsに変更する関数を一時的に追加

This commit is contained in:
misyaguziya
2024-02-12 22:29:32 +09:00
parent b46a9d5664
commit c34aac28f6
2 changed files with 12 additions and 2 deletions

View File

@@ -8,6 +8,10 @@ if __name__ == "__main__":
splash.showSplash()
from config import config
# version 2.2.0からweightフォルダをweightsに変更する
from utils import renameWeightFolder
renameWeightFolder(config.PATH_LOCAL)
from models.translation.translation_utils import downloadCTranslate2Weight
if config.USE_TRANSLATION_FEATURE is True:
downloadCTranslate2Weight(config.PATH_LOCAL, config.CTRANSLATE2_WEIGHT_TYPE, splash.updateDownloadProgress)

View File

@@ -1,5 +1,5 @@
from typing import Union
from os import path as os_path
from os import path as os_path, rename as os_rename
from PIL.Image import open as Image_open
def getImageFile(file_name):
@@ -50,3 +50,9 @@ def isUniqueStrings(unique_strings:Union[str, list], input_string:str, require=F
else:
# If require is False, check if unique strings are used exactly once
return all(count == 1 for count in counts)
# path先のweightフォルダがある場合にはそのフォルダ名をweightsに変更する
def renameWeightFolder(path):
weight_path = os_path.join(path, "weight")
if os_path.exists(weight_path):
os_rename(weight_path, os_path.join(path, "weights"))