vslinx commited on
Commit
da29efb
·
1 Parent(s): fef7f83

added sam model download

Browse files
Files changed (1) hide show
  1. windows-nvidia.bat +17 -0
windows-nvidia.bat CHANGED
@@ -171,6 +171,23 @@ if not exist "%workflowFolder%" (
171
  mkdir "%workflowFolder%"
172
  )
173
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
174
  echo Checking workflow files...
175
 
176
  REM Define workflow filenames and URLs
 
171
  mkdir "%workflowFolder%"
172
  )
173
 
174
+ REM -------------------------------
175
+ REM Download SAM model (Segment Anything)
176
+ REM -------------------------------
177
+ set "samFolder=%comfyPath%\models\sams"
178
+ if not exist "%samFolder%" (
179
+ mkdir "%samFolder%"
180
+ )
181
+
182
+ set "samFile=%samFolder%\sam_vit_b_01ec64.pth"
183
+ if exist "%samFile%" (
184
+ echo - sam_vit_b_01ec64.pth already exists. Skipping download.
185
+ ) else (
186
+ echo Downloading SAM model...
187
+ %SystemRoot%\System32\curl.exe -L -o "%samFile%" ^
188
+ https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth
189
+ )
190
+
191
  echo Checking workflow files...
192
 
193
  REM Define workflow filenames and URLs