tortoise-base / src /utils.py

Commit History

actually accumulate derivatives when estimating milestones and final loss by using half of the log
b8a620e

ecker commited on

oops v2
35225a3

ecker commited on

5 hour sleep brained
b5e9899

ecker commited on

oops
cd8702a

ecker commited on

reordered things so it uses fresh data and not last-updated data
d312019

ecker commited on

added '''estimating''' iterations until milestones (lr=[1, 0.5, 0.1] and final lr, very, very inaccurate because it uses instantaneous delta lr, I'll need to do a riemann sum later
ce3866d

ecker commited on

forgot to try and have it try and auto-detect for openai/whisper when no language is specified
1316331

ecker commited on

added option to set worker size in training config generator (because the default is overkill), for whisper transcriptions, load a specialized language model if it exists (for now, only english), output transcription to web UI when done transcribing
3e220ed

ecker commited on

sloppy fix to actually kill children when using multi-GPU distributed training, set GPU training count based on what CUDA exposes automatically so I don't have to keep setting it to 2
5026d93

ecker commited on

forgot to add 'bs / gradient accum < 2 clamp validation logic
1a9d159

ecker commited on

renamed mega batch factor to an actual real term: gradient accumulation factor, fixed halting training not actually killing the training process and freeing up resources, some logic cleanup for gradient accumulation (so many brain worms and wrong assumptions from testing on low batch sizes) (read the training section in the wiki for more details)
df24827

ecker commited on

fixed user inputted LR schedule not actually getting used (oops)
6d5e1e1

ecker commited on

auto-suggested voice chunk size is based on the total duration of the voice files divided by 10 seconds, added setting to adjust the auto-suggested division factor (a really oddly worded one), because I'm sure people will OOM blindly generating without adjusting this slider
6d8c2dd

ecker commited on

oops
e1f3ffa

ecker commited on

validated whispercpp working, fixed args.listen not being saved due to brainworms
9fb4aa7

ecker commited on

added option to specify using BigVGAN as the vocoder for mrq/tortoise-tts
740b558

ecker commited on

oops
68f4858

ecker commited on

experimental multi-gpu training (Linux only, because I can't into batch files)
e859a7c

ecker commited on

added button to just load a training set's loss information, added installing broncotc/bitsandbytes-rocm when running setup-rocm.sh
c956d81

ecker commited on

added loading/saving of voice latents by model hash, so no more needing to manually regenerate every time you change models
534a761

ecker commited on

oops
5a41db9

ecker commited on

leverage tensorboard to parse tb_logger files when starting training (it seems to give a nicer resolution of training data, need to see about reading it directly while training)
b989123

ecker commited on

added new training tunable: loss_text_ce_loss weight, added option to specify source model in case you want to finetune a finetuned model (for example, train a Japanese finetune on a large dataset, then finetune for a specific voice, need to truly validate if it produces usable output), some bug fixes that came up for some reason now and not earlier
c2726fa

ecker commited on

oops
5037752

ecker commited on

added to embedded metadata: datetime, model path, model hash
787b448

ecker commited on

show different losses, rewordings
81eb58f

ecker commited on

oops
fda4715

ecker commited on

added graph to chart loss_gpt_total rate, added option to prune X number of previous models/states, something else
bc0d9ab

ecker commited on

I don't remember.
6925ec7

ecker commited on

Added option to disable bitsandbytesoptimizations for systems that do not support it (systems without a Turing-onward Nvidia card), saves use of float16 and bitsandbytes for training into the config json
9255397

ecker commited on

actually fixed the training output text parser
aafeb9f

ecker commited on

oops, epoch increments twice
65329db

ecker commited on

csome adjustments to the training output parser, now updates per iteration for really large batches (like the one I'm doing for a dataset size of 19420)
8b4da29

ecker commited on

fixed some files not copying for bitsandbytes (I was wrong to assume it copied folders too), fixed stopping generating and training, some other thing that I forgot since it's been slowly worked on in my small free times
d5d8821

ecker commited on

ops
2104dbd

ecker commited on

finally added model refresh button, also searches in the training folder for outputted models so you don't even need to copy them
f6d0b66

ecker commited on

god i finally found some time and focus: reworded print/save freq per epoch => print/save freq (in epochs), added import config button to reread the last used settings (will check for the output folder's configs first, then the generated ones) and auto-grab the last resume state (if available), some other cleanups i genuinely don't remember what I did when I spaced out for 20 minutes
1e0fec4

ecker commited on

forgot to mult by batch size
7d1220e

ecker commited on

fixed the brain worm discrepancy between epochs, iterations, and steps
487f2eb

ecker commited on

oops
1cbcf14

ecker commited on

huge success
225dee2

ecker commited on

how did this revert...
526a430

ecker commited on

oops
93b061f

ecker commited on

properly placed the line toe xtract starting iteration
c4b41e0

ecker commited on

oops
fefc7ab

ecker commited on

clamp batch size to sample count when generating for the sickos that want that, added setting to remove non-final output after a generation, something else I forgot already
9e64dad

ecker commited on

explicitly use python3 because some OSs will not have python alias to python3, allow batch size 1
f119993

ecker commited on

Added very experimental float16 training for cards with not enough VRAM (10GiB and below, maybe) \!NOTE\! this is VERY EXPERIMETNAL, I have zero free time to validate it right now, I'll do it later
8a1a48f

ecker commited on

wrap checking for metadata when adding a voice in case it throws an error
ed2cf9f

ecker commited on

fixes
b6f7aa6

ecker commited on