AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 08:15:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update download-model.bat

Browse Source 
Added variables for model selection and text only mode.
Updated virtual environment handling to use Micromamba.
This commit is contained in:
jllllll 2023-03-24 18:15:24 -05:00 committed by GitHub
parent 24870e51ed
commit 2604e3f7ac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
download-model.bat
View File

@ -1,9 +1,28 @@
@echo off
set INSTALL_ENV_DIR=%cd%\installer_files\env
set PATH=%INSTALL_ENV_DIR%;%INSTALL_ENV_DIR%\Library\bin;%INSTALL_ENV_DIR%\Scripts;%INSTALL_ENV_DIR%\Library\usr\bin;%PATH%
call conda activate
cd text-generation-webui
python download-model.py
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