mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-31 22:50:15 +01:00
Update download-model.bat
Added variables for model selection and text only mode. Updated virtual environment handling to use Micromamba.
This commit is contained in:
parent
24870e51ed
commit
2604e3f7ac
@ -1,9 +1,28 @@
|
|||||||
@echo off
|
@echo off
|
||||||
|
|
||||||
set INSTALL_ENV_DIR=%cd%\installer_files\env
|
SET TextOnly=False &REM True or False for Text only mode
|
||||||
set PATH=%INSTALL_ENV_DIR%;%INSTALL_ENV_DIR%\Library\bin;%INSTALL_ENV_DIR%\Scripts;%INSTALL_ENV_DIR%\Library\usr\bin;%PATH%
|
SET ModelName="chansung/alpaca-lora-13b" &REM HuggingFace model repo eg. "facebook/opt-1.3b"
|
||||||
call conda activate
|
|
||||||
cd text-generation-webui
|
|
||||||
python download-model.py
|
|
||||||
|
|
||||||
|
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
|
pause
|
||||||
|
Loading…
Reference in New Issue
Block a user