✨[Feature] utils : weightフォルダ名をweightsに変更する関数を一時的に追加
This commit is contained in:
4
main.py
4
main.py
@@ -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)
|
||||
|
||||
8
utils.py
8
utils.py
@@ -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"))
|
||||
Reference in New Issue
Block a user