diff --git a/download_resource.bat b/download_resource.bat index a93fff5..0e868c7 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -112,7 +112,7 @@ SET /A EXISTING_FILES=0 SET MISSING_LIST= REM Neural Haircut Diffusion Prior 检查 -IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\model.pt" ( +IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" ( echo [√] Neural Haircut Diffusion Prior已存在 SET /A EXISTING_FILES+=1 ) ELSE ( @@ -122,7 +122,7 @@ IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\model.p ) REM Neural Haircut Strand Prior 检查 -IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\model.pt" ( +IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" ( echo [√] Neural Haircut Strand Prior已存在 SET /A EXISTING_FILES+=1 ) ELSE ( @@ -256,14 +256,14 @@ echo 正在验证下载文件... cd %PROJECT_DIR% SET VERIFY_FAILED=0 -IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\model.pt" ( +IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" ( echo [×] 缺少Neural Haircut Diffusion Prior SET /A VERIFY_FAILED+=1 ) ELSE ( echo [√] Neural Haircut Diffusion Prior验证成功 ) -IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\model.pt" ( +IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" ( echo [×] 缺少Neural Haircut Strand Prior SET /A VERIFY_FAILED+=1 ) ELSE ( diff --git a/install.bat b/install.bat index 2f16f88..470247d 100644 --- a/install.bat +++ b/install.bat @@ -154,9 +154,9 @@ IF NOT EXIST "%PROJECT_DIR%\resource" ( echo ├── NeuralHaircut/ echo │ ├── pretrained_models/ echo │ │ ├── diffusion_prior/ - echo │ │ │ └── model.pt + echo │ │ │ └── dif_ckpt.pt echo │ │ └── strand_prior/ - echo │ │ └── model.pt + echo │ │ └── strand_ckpt.pt echo │ └── PIXIE/ echo │ └── pixie_data/ echo ├── Matte-Anything/