This commit is contained in:
Jeffreytsai1004 2025-02-16 16:59:52 +08:00
parent 15438b139f
commit 818304772b
2 changed files with 6 additions and 6 deletions

View File

@ -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 (

View File

@ -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/