TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #42 from hayden-fr/hotfix

Browse Source 
fix: Cross-device movement
This commit is contained in:
Hayden
2024-11-04 12:05:28 +08:00
committed by GitHub
parent 79b106d986 da1ec3a52c
commit 1f4c55ab89
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
py/utils.py
View File

@@ -273,7 +273,7 @@ def rename_model(model_path: str, new_model_path: str):
os.makedirs(new_model_dirname) os.makedirs(new_model_dirname)
# move model # move model
os.rename(model_path, new_model_path) shutil.move(model_path, new_model_path)
# move preview # move preview
previews = get_model_all_images(model_path) previews = get_model_all_images(model_path)
@@ -288,14 +288,14 @@ def rename_model(model_path: str, new_model_path: str):
new_model_dirname, new_model_name + ".preview" + preview_ext new_model_dirname, new_model_name + ".preview" + preview_ext
) )
) )
os.rename(preview_path, new_preview_path) shutil.move(preview_path, new_preview_path)
# move description # move description
description = get_model_description_name(model_path) description = get_model_description_name(model_path)
description_path = os.path.join(model_dirname, description) description_path = os.path.join(model_dirname, description)
if os.path.isfile(description_path): if os.path.isfile(description_path):
new_description_path = os.path.join(new_model_dirname, f"{new_model_name}.md") new_description_path = os.path.join(new_model_dirname, f"{new_model_name}.md")
os.rename(description_path, new_description_path) shutil.move(description_path, new_description_path)
import pickle import pickle