fix: weights folder

This commit is contained in:
源文雨
2023-08-30 00:27:23 +08:00
parent 417c1e58a1
commit 1890584496
5 changed files with 6 additions and 6 deletions

View File

@@ -36,7 +36,7 @@ shutil.rmtree("%s/runtime/Lib/site-packages/infer_pack" % (now_dir), ignore_erro
shutil.rmtree("%s/runtime/Lib/site-packages/uvr5_pack" % (now_dir), ignore_errors=True)
os.makedirs(tmp, exist_ok=True)
os.makedirs(os.path.join(now_dir, "logs"), exist_ok=True)
os.makedirs(os.path.join(now_dir, "weights"), exist_ok=True)
os.makedirs(os.path.join(now_dir, "assets/weights"), exist_ok=True)
warnings.filterwarnings("ignore")
torch.manual_seed(114514)