Commit Graph

157 Commits

Author SHA1 Message Date
harubaru 4f9070af3c use penultimate layer
from NAI blogpost and https://arxiv.org/pdf/2205.11487.pdf
2022-10-12 17:55:56 -07:00
harubaru e4736c11f5 add initial config 2022-10-12 17:45:44 -07:00
harubaru a4628125fe
fix pruning script yet again 2022-09-27 19:50:55 -07:00
harubaru aa91eaf228
gotta go fast 2022-09-24 01:13:36 -07:00
harubaru ae05402bc6
speedup 2022-09-23 15:26:58 -07:00
harubaru de1470f43b fix downloader 2022-09-22 19:49:08 -07:00
harubaru d0ec5be718
add centercropping 2022-09-22 19:45:36 -07:00
harubaru 5b169173d5
fix config 2022-09-22 17:44:19 -07:00
harubaru c5f2775beb Add alternate dataloader 2022-09-22 14:56:27 -07:00
harubaru 2e69358d50 Add random tag order during training 2022-09-21 04:57:17 -07:00
harubaru b080f33115 Add variable aspect ratio during training 2022-09-21 04:47:39 -07:00
harubaru c690d005bc fix cropping 2022-09-20 22:24:04 -07:00
harubaru 01b9440d50 dataloader overhaul 2022-09-20 22:00:32 -07:00
harubaru 3546ae9b47 add 8 gpu config 2022-09-20 11:34:06 -07:00
harubaru 106e611306
Update README 2022-09-18 22:36:49 -07:00
harubaru 3020278dd9
Merge pull request #17 from derfred/installfixes
docs for executing
2022-09-17 08:11:30 -07:00
Frederik Fix 7cf4130668 make the execution work 2022-09-17 08:55:56 +02:00
Frederik Fix b664c86e89
Update executing.md 2022-09-15 19:51:56 +02:00
Frederik Fix 0cfa39a373
Update train.sh 2022-09-15 19:46:52 +02:00
Frederik Fix ab67872afa
Update requirements.txt 2022-09-15 19:44:08 +02:00
Frederik Fix af00e49aa8
Update dataset.md 2022-09-15 19:43:15 +02:00
Frederik Fix 98c6e0c38b
Create executing.md 2022-09-15 18:42:28 +02:00
Frederik Fix 90a580f684
Update README.md 2022-09-15 18:40:47 +02:00
Frederik Fix 0e2fd255ba
wrong arg 2022-09-15 18:39:05 +02:00
harubaru bbcd956746
Merge pull request #16 from derfred/doc-fixes
Doc fixes
2022-09-14 12:18:28 -07:00
Frederik Fix 40640acc83
use the directory hard coded 2022-09-14 21:13:48 +02:00
Frederik Fix 9113f9155a
some missing info 2022-09-14 21:10:55 +02:00
Frederik Fix d6a3ca65ce
fix extract command 2022-09-14 20:09:08 +02:00
Frederik Fix b4cc0d8279
download directories 2022-09-14 19:59:42 +02:00
harubaru f6c06933ba
Merge pull request #14 from chavinlo/patch-3
moved scripts and added nsfw extractors
2022-09-11 22:55:03 -07:00
chavinlo c7c892ae4e moved scripts and added nsfw extractors 2022-09-11 11:44:55 -05:00
harubaru 7194a68b44
Merge pull request #13 from chavinlo/patch-2
Added option to extract only specific ratings and arguments from cli
2022-09-10 18:39:55 -07:00
Carlos Chavez a532563a71
Merge branch 'harubaru:main' into patch-2 2022-09-10 19:40:10 -05:00
chavinlo 7f901bf252 removed string import 2022-09-10 19:39:49 -05:00
chavinlo f07fb01490 Added rating selection and now works 2022-09-10 19:38:22 -05:00
harubaru bd6986fcc6
remove link 2022-09-09 15:07:18 -07:00
harubaru 3a84dc1973
add data download files 2022-09-09 10:42:56 -07:00
chavinlo 4904ba4db3 in the works 2022-09-09 03:49:52 -05:00
harubaru 41c864dce2
Merge pull request #12 from chavinlo/patch-1
Fix Images in documentation and add mirrors of weights
2022-09-09 01:18:37 -07:00
chavinlo 6da7d93133 relink 2022-09-09 02:47:39 -05:00
chavinlo 6b35ad0c79 Update README.md 2022-09-09 02:23:08 -05:00
chavinlo 9912d3c3a6 docs 2022-09-09 02:22:16 -05:00
harubaru df8761695b
redo training launcher 2022-09-08 18:46:23 -07:00
harubaru 952bdc04df
Merge pull request #10 from throwaway-mezzo-mix/patch-1
fix tag replace typo in danbooru21_extract.py
2022-09-08 18:15:36 -07:00
throwaway-mezzo-mix b9452bc0fd
fix tag replace typo in danbooru21_extract.py
1boy was being replaced with 'one girl', now it correctly gets replaced with 'one boy'.
2022-09-08 20:52:38 +02:00
harubaru d283e04b8b
remove outputs from cudacallback 2022-09-07 03:51:04 -07:00
harubaru f7bb9b7bbc fix validation 2022-09-06 23:33:12 -07:00
harubaru 51666f070c
fix link 2022-09-05 18:41:08 -07:00
harubaru 8b08d8ebd9
fix link 2022-09-05 18:40:40 -07:00
harubaru cb417c437e
fix link 2022-09-05 18:39:29 -07:00