diff --git a/download_resource.bat b/download_resource.bat
index 764e3a6..ebce497 100644
--- a/download_resource.bat
+++ b/download_resource.bat
@@ -98,6 +98,7 @@ REM 创建resource目录结构
 mkdir resource 2>nul
 cd resource
 mkdir NeuralHaircut 2>nul
+mkdir NeuralHaircut\pretrained_models 2>nul
 mkdir Matte-Anything 2>nul
 mkdir openpose 2>nul
 mkdir hyperIQA 2>nul
@@ -206,8 +207,14 @@ cd resource\NeuralHaircut
 echo 正在下载Neural Haircut预训练文件夹...
 python -m gdown --folder "%NEURAL_HAIRCUT_FOLDER_URL%"
 
+REM 更新 diffusion_prior 模型
+cd pretrained_models\diffusion_prior
+echo 正在更新Neural Haircut Diffusion Prior模型...
+python -m gdown "%NEURAL_HAIRCUT_MODEL_URL%"
+cd ..\..
+
 REM 下载 PIXIE 文件
-cd NeuralHaircut\PIXIE
+cd PIXIE
 IF EXIST "pixie_data.tar" (
     echo PIXIE模型已存在，跳过下载...
 ) ELSE (
diff --git a/install.bat b/install.bat
index 90dc3cd..15c0d59 100644
--- a/install.bat
+++ b/install.bat
@@ -164,10 +164,11 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
     echo.
     echo resource/
     echo ├── NeuralHaircut/
-    echo │   ├── diffusion_prior/
-    echo │   │   └── dif_ckpt.pt
-    echo │   ├── strand_prior/
-    echo │   │   └── strand_ckpt.pt
+    echo │   ├── pretrained_models/
+    echo │   │   ├── diffusion_prior/
+    echo │   │   │   └── dif_ckpt.pt
+    echo │   │   └── strand_prior/
+    echo │   │       └── strand_ckpt.pt
     echo │   └── PIXIE/
     echo │       └── pixie_data/
     echo ├── Matte-Anything/
@@ -186,13 +187,13 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
 )
 
 REM 检查模型文件是否存在
-IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\diffusion_prior\dif_ckpt.pt" (
+IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\dif_ckpt.pt" (
     echo ERROR: Neural Haircut Diffusion Prior model not found
     echo Please run download_resource.bat first
     exit /b 1
 )
 
-IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\strand_prior\strand_ckpt.pt" (
+IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\strand_prior\strand_ckpt.pt" (
     echo ERROR: Neural Haircut Strand Prior model not found
     echo Please run download_resource.bat first
     exit /b 1
@@ -239,8 +240,9 @@ echo 正在复制模型文件...
 
 REM 创建必要的目录
 mkdir "%PROJECT_DIR%\ext\NeuralHaircut" 2>nul
-mkdir "%PROJECT_DIR%\ext\NeuralHaircut\diffusion_prior" 2>nul
-mkdir "%PROJECT_DIR%\ext\NeuralHaircut\strand_prior" 2>nul
+mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models" 2>nul
+mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior" 2>nul
+mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\strand_prior" 2>nul
 mkdir "%PROJECT_DIR%\ext\PIXIE" 2>nul
 mkdir "%PROJECT_DIR%\ext\Matte-Anything\pretrained" 2>nul
 mkdir "%PROJECT_DIR%\ext\openpose\models\pose\coco" 2>nul