Update download_resource.bat

This commit is contained in:
Jeffreytsai1004 2025-02-16 17:46:12 +08:00
parent 4376af31d3
commit b9348b6033

View File

@ -204,16 +204,24 @@ IF DEFINED HTTP_PROXY (
REM 下载 Neural Haircut 文件
cd resource\NeuralHaircut
echo 正在下载Neural Haircut预训练文件夹...
python -m gdown --folder "%NEURAL_HAIRCUT_FOLDER_URL%"
REM 更新 diffusion_prior 模型
mkdir pretrained_models 2>nul
mkdir pretrained_models\diffusion_prior 2>nul
mkdir pretrained_models\strand_prior 2>nul
REM 下载 diffusion_prior 模型
cd pretrained_models\diffusion_prior
echo 正在更新Neural Haircut Diffusion Prior模型...
python -m gdown "%NEURAL_HAIRCUT_MODEL_URL%"
echo 正在下载Neural Haircut Diffusion Prior模型...
python -m gdown "1OBR0Vzb_w5SNc1jmoScA6L8SqVniJGOr" -O dif_ckpt.pt
cd ..
REM 下载 strand_prior 模型
cd strand_prior
echo 正在下载Neural Haircut Strand Prior模型...
python -m gdown "1DESwUb-nsmi38VCDvnBwpd9kjcWONNT6" -O strand_ckpt.pt
cd ..\..
REM 下载 PIXIE 文件
mkdir PIXIE 2>nul
cd PIXIE
IF EXIST "pixie_data.tar" (
echo PIXIE模型已存在跳过下载...