Skip to content

Commit bf1b36c

Browse files
authored
Update and rename 1-click-run.bat to start_cuda.bat
1 parent 0cfa985 commit bf1b36c

File tree

2 files changed

+21
-20
lines changed

2 files changed

+21
-20
lines changed

1-click-run.bat

Lines changed: 0 additions & 20 deletions
This file was deleted.

start_cuda.bat

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
@echo off
2+
call message.bat
3+
pause
4+
5+
python -m pip install virtualenv
6+
7+
call install.bat
8+
9+
set PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.8,max_split_size_mb:512
10+
set CUDA_MODULE_LOADING=LAZY
11+
set CUDA_PATH=venv\Lib\site-packages\torch\lib
12+
13+
call venv\Scripts\activate.bat
14+
python run.py --execution-threads 8 --execution-provider cuda --max-memory 16 --video-encoder libx264 --frame-processor face_swapper
15+
pause
16+
17+
REM Упаковано и собрано телеграм каналом Neutogen News: https://t.me/neurogen_news
18+
REM --gpu-threads N - Количество потоков для вашей видеокарты. Слишком большое значение может вызвать ошибки или наоборот, снизить производительность. 4 потока потребляют примерно 5.5-6 Gb VRAM, 8 потоков - 10 Gb VRAM, но пиковое потребление бывает выше.
19+
REM --max-memory 8 - сколько вы готовы выделить оперативной памяти в гигабайтах
20+
REM --frame-processor - выбор режима обработки. face_swapper - замена лица, face_enhancer - улучшение уже измененного лица через апскейлер
21+
REM --execution-provider cuda - Наш бэкенд. Если у вас Nvidia - ставьте cuda, если Nvidia RTX 4000 - можете попробовать указать tensorrt.

0 commit comments

Comments
 (0)