Update download_resource.bat
This commit is contained in:
parent
4376af31d3
commit
b9348b6033
@ -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模型已存在,跳过下载...
|
||||
|
Loading…
Reference in New Issue
Block a user