Hugging Face's logo

aka7774
/
frog_bench

Model card Files
xet
Community
frog_bench
File size: 3,857 Bytes 
5dd68ef
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
b5e9ff4
5dd68ef
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
 
@echo off
rem ### frog train benchmark ###

rem bypass Install CUDA Toolkit
rem SET PATH=%PATH%;C:\SD\stable-diffusion-webui\venv\Lib\site-packages\torch\lib

rem Path to
rem SET PYTHON=C:\SD\Python310\Python.exe
rem SET GIT=C:\SD\PortableGit\bin\git.exe
SET PYTHON=python
SET GIT=git

rem VERS
rem  1. torch==1.12.1+cu116 xformers-0.0.14.dev0
rem  2. torch==1.13.1+cu117 xformers-0.0.16rc425
SET VERS=1

rem MODE
rem  1. install+train+inference
rem  2. train+inference
rem  3. inference
SET MODE=1

rem BATCH_SIZE=2 IF VRAM < 10GB
SET BATCH_SIZE=4

rem #############################

if %MODE% == 1 (
rem latest sd-scripts
%GIT% clone https://github.com/kohya-ss/sd-scripts.git
)
cd sd-scripts

if %MODE% == 1 (
rem frog and SDv1.5
%GIT% clone https://huggingface.co/aka7774/frog_bench.git

%PYTHON% -m venv venv
)

rem venv activate
SET PYTHON=%cd%\venv\Scripts\Python.exe
SET PIP=%cd%\venv\Scripts\pip.exe
SET ACCELERATE=%cd%\venv\Scripts\accelerate.exe

if %MODE% == 1 (

if %VERS% == 1 (
"%PYTHON%" -m pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
"%PYTHON%" -m pip install --upgrade -r requirements.txt
"%PYTHON%" -m pip install -U -I --no-deps https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl
)

if %VERS% == 2 (
"%PYTHON%" -m pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
"%PYTHON%" -m pip install --upgrade -r requirements.txt
"%PYTHON%" -m pip install -U -I --no-deps xformers==0.0.16rc425
)

copy /y .\bitsandbytes_windows\*.dll .\venv\Lib\site-packages\bitsandbytes\
copy /y .\bitsandbytes_windows\cextension.py .\venv\Lib\site-packages\bitsandbytes\cextension.py
copy /y .\bitsandbytes_windows\main.py .\venv\Lib\site-packages\bitsandbytes\cuda_setup\main.py
)

if %MODE% leq 2 (

call :GetStartTime

%ACCELERATE% launch ^
 --num_cpu_threads_per_process 4 ^
 --num_processes 1 ^
 --num_machines 1 ^
 --dynamo_backend no ^
 --mixed_precision fp16 ^
 train_network.py ^
 --pretrained_model_name_or_path=frog_bench/model/v1-5-pruned-pruned-fp16.safetensors ^
 --train_data_dir=frog_bench/train ^
 --reg_data_dir=frog_bench/reg ^
 --prior_loss_weight=1.0 ^
 --resolution 512 ^
 --output_dir=lora_output ^
 --output_name=cjgg_frog ^
 --train_batch_size=%BATCH_SIZE% ^
 --learning_rate=1e-4 ^
 --max_train_epochs 4 ^
 --use_8bit_adam ^
 --xformers ^
 --mixed_precision=fp16 ^
 --save_precision=fp16 ^
 --seed 42 ^
 --save_model_as=safetensors ^
 --max_data_loader_n_workers=1 ^
 --network_module=networks.lora ^
 --network_dim=4 ^
 --training_comment="activate by usu frog"

call :GetEndTime

)

%PYTHON% gen_img_diffusers.py ^
 --ckpt frog_bench/model/v1-5-pruned-pruned-fp16.safetensors ^
 --n_iter 1 ^
 --scale 7.5 ^
 --steps 40 ^
 --outdir txt2img ^
 --xformers ^
 --W 512 ^
 --H 512 ^
 --fp16 ^
 --sampler k_euler_a ^
 --network_module networks.lora ^
 --network_weights lora_output\cjgg_frog.safetensors ^
 --network_mul 1.0 ^
 --max_embeddings_multiples 3 ^
 --clip_skip 1 ^
 --batch_size 1 ^
 --images_per_prompt 1 ^
 --prompt "usu frog"

if %MODE% leq 2 (
call :PutTime
)

pause >nul
goto EOL

:GetStartTime
set T=%TIME: =0%
set H=%T:~0,2%
set M=%T:~3,2%
set S=%T:~6,2%
set C=%T:~9,2%
set /a H=1%H%-100,M=1%M%-100,S=1%S%-100,C=1%C%-100
exit /b 0

:GetEndTime
set T1=%TIME: =0%
set H1=%T1:~0,2%
set M1=%T1:~3,2%
set S1=%T1:~6,2%
set C1=%T1:~9,2%
set /a H1=1%H1%-100,M1=1%M1%-100,S1=1%S1%-100,C1=1%C1%-100

set /a H2=H1-H,M2=M1-M
if %M2% LSS 0 set /a H2=H2-1,M2=M2+60
 set /a S2=S1-S
if %S2% LSS 0 set /a M2=M2-1,S2=S2+60
set /a C2=C1-C
if %C2% LSS 0 set /a S2=S2-1,C2=C2+100
if %C2% LSS 10 set C2=0%C2%
exit /b 0

:PutTime
rem echo %T%
rem echo %T1%
echo %H2%h%M2%m%S2%.%C2%s
echo %H2%h%M2%m%S2%.%C2%s>result.txt
exit /b 0

:EOL