Skip to content

Commit

Permalink
Merge pull request #11 from e1113633/launch_script
Browse files Browse the repository at this point in the history
Add files via upload
  • Loading branch information
aqborromeo authored Nov 2, 2023
2 parents 2904022 + 04be2e1 commit 64c0c15
Show file tree
Hide file tree
Showing 2 changed files with 57 additions and 0 deletions.
24 changes: 24 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
diffusers[torch]==0.21.4
flask==2.2.2
torch==2.1.0+cu118
torchvision==0.16.0+cu118
ipykernel==6.25.0
voluptuous==0.13.1
matplotlib==3.8.0
rich==13.6.0
werkzeug==2.2.3
ipywidgets==8.0.4
pip==20.3
accelerate==0.18.0
albumentations==1.3.0
diffusers==0.21.4
polars==0.19.7
opencv-python==4.7.0.68
omegaconf==2.3.0
pytorch-lightning==1.9.0
einops==0.6.0
torchinfo==1.8.0
transformers==4.26.0
tensorboard==2.12.0
safetensors==0.3.1
Flask-Cors==1.10.3
33 changes: 33 additions & 0 deletions webgui.bat
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
@echo off

set PYTHON_VER=3.9.18
set PARENT_DIR=%CD%

:: Check if Python version meets the recommended version
python --version 2>nul | findstr /b /c:"Python %PYTHON_VER%" >nul
if errorlevel 1 (
echo Warning: Python version %PYTHON_VER% is recommended.
)


IF NOT EXIST venv (
echo Creating venv...
python -m venv venv
)


:: Create the directory if it doesn't exist
mkdir ".\logs\setup" > nul 2>&1

:: install requirements
python.exe -m pip install -r requirements.txt

:: change directory to main
chdir ".\SystemCode\src\main"
start cmd /k "flask run"

chdir /d %PARENT_DIR%


:: Deactivate the virtual environment
call .\venv\Scripts\deactivate.bat

0 comments on commit 64c0c15

Please sign in to comment.