mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-09-21 02:55:10 +02:00
2604e3f7ac
Added variables for model selection and text only mode. Updated virtual environment handling to use Micromamba.
28 lines
742 B
Batchfile
28 lines
742 B
Batchfile
@echo off
|
|
|
|
SET TextOnly=False &REM True or False for Text only mode
|
|
SET ModelName="chansung/alpaca-lora-13b" &REM HuggingFace model repo eg. "facebook/opt-1.3b"
|
|
|
|
cd /D "%~dp0"
|
|
|
|
set MAMBA_ROOT_PREFIX=%cd%\installer_files\mamba
|
|
set INSTALL_ENV_DIR=%cd%\installer_files\env
|
|
|
|
if not exist "%MAMBA_ROOT_PREFIX%\Scripts\activate.bat" (
|
|
call "%MAMBA_ROOT_PREFIX%\micromamba.exe" shell hook >nul 2>&1
|
|
)
|
|
call "%MAMBA_ROOT_PREFIX%\condabin\mamba_hook.bat" || ( echo MicroMamba hook not found. && goto end )
|
|
call micromamba activate "%INSTALL_ENV_DIR%" || goto end
|
|
|
|
cd text-generation-webui || goto end
|
|
goto %TextOnly%
|
|
|
|
:False
|
|
call python download-model.py %ModelName%
|
|
goto end
|
|
|
|
:True
|
|
call python download-model.py %ModelName% --text-only
|
|
|
|
:end
|
|
pause
|