Commit Graph

  • b09328069e actually do CFG sampling for base AR+NAR tasks mrq 2024-11-12 13:42:39 -0600
  • 2495a7ef67 Fixed STT in the web UI mrq 2024-11-12 12:49:53 -0600
  • 8927bad7bc actually fixed rep pen (for ar and nar, it seems to help with nar unmasking) mrq 2024-11-11 21:40:19 -0600
  • ec92613847 actually pass input prompt length size to inference mrq 2024-11-11 20:39:48 -0600
  • b1df6a7bed reverted rep pen sampler due to a regression mrq 2024-11-11 20:35:08 -0600
  • b1f4db39c8 threw in CFG sampling for normal model as well to experiment with mrq 2024-11-11 20:27:38 -0600
  • 2f56696506 overhauled inference/sampler kwargs to stop being a bloated mess mrq 2024-11-11 20:21:16 -0600
  • 354f8e059d store dataset hash alongside state dict so it can be ignored if mismatched mrq 2024-11-11 18:16:56 -0600
  • f7b8b1e825 dropped subtrain dataloader since its useless to duplicate mrq 2024-11-11 17:00:49 -0600
  • cf9df71f2c use homwbrewed caching system for dataloader paths / durations (I'm pretty sure I am now triggering OOM killers with my entire dataset used) mrq 2024-11-11 16:32:08 -0600
  • a748e223ce tweaks mrq 2024-11-11 12:40:41 -0600
  • 48490757da fixes mrq 2024-11-10 20:37:50 -0600
  • 9def34cd66 lol mrq 2024-11-10 12:48:41 -0600
  • 9cb0b6901b unified nar.py into ar_nar.py mrq 2024-11-10 12:19:48 -0600
  • a9d2faf2d7 all I can do now until I wait for the model to (re)train for pure NAR mrq 2024-11-09 22:57:34 -0600
  • ad7e290a5e ugh (ROCm seems to silently clamp any token value >= logits.shape[-1] for loss calculation, while cuda will throw an assert, making it hard to find this dumb fuckup) mrq 2024-11-09 19:40:02 -0600
  • 943fe70c10 I don't know why this fixes an assert thrown but it does mrq 2024-11-09 19:04:13 -0600
  • f50d92ba6c Almost made a mistake mrq 2024-11-09 18:12:54 -0600
  • c6a38693a2 This better work mrq 2024-11-09 18:04:59 -0600
  • 8b3d1cf70a Something's Wrong mrq 2024-11-09 15:07:43 -0600
  • dcd5fecff3 some cleanup while I wait for the NAR-len to train to an acceptable state (currently it performs okay, but only on audo after 3 seconds or so) mrq 2024-11-09 12:12:46 -0600
  • 69b0b3b854 set timestep tensor to whatever the time embedding's dtype is because it'll gripe under amp mrq 2024-11-09 00:11:16 -0600
  • 5a09a5f6e9 I forgot about the time embedding... mrq 2024-11-08 22:46:26 -0600
  • 811b15d280 I suppose I just have a shit training method since the sampler is as solid as I can get it............... mrq 2024-11-08 22:05:41 -0600
  • 13b54953bd agony mrq 2024-11-08 13:34:39 -0600
  • c127c4e488 'borrowed' a sampling scheduler for NAR-len's RVQ level 0 (better than before, but still not good enough) mrq 2024-11-07 21:19:14 -0600
  • e108c54daf new NAR-len training paradigm...... mrq 2024-11-07 11:32:11 -0600
  • ed174c589e ugh mrq 2024-11-07 09:19:21 -0600
  • d13ab00ad8 one more note mrq 2024-11-07 09:11:21 -0600
  • 5698188824 あたしって、ほんとバカ mrq 2024-11-07 09:10:18 -0600
  • 77ff23e319 repeat extend the prom to fill the initial tokens for nar-len (it somewhat works, the model just needs to train more) mrq 2024-11-06 23:29:53 -0600
  • a3bc26f7ec ugh mrq 2024-11-06 23:16:28 -0600
  • d606a693ff eval fix for nar-len mrq 2024-11-06 23:14:16 -0600
  • 105ed51159 I guess I'll fall for the NAR-len meme again (I don't know where my previous weights are, so I need to train it again to test something) mrq 2024-11-06 19:17:12 -0600
  • bcabde3454 more notes mrq 2024-11-06 13:51:28 -0600
  • bfc5e1d723 agony mrq 2024-11-05 22:30:49 -0600
  • aefe8fcdad UGH mrq 2024-11-05 22:13:58 -0600
  • 556d9db0d5 web UI support for HF ZeroGPU mrq 2024-11-05 21:38:02 -0600
  • e58a9469a3 move layerskip to experimental settings....... mrq 2024-11-05 20:37:06 -0600
  • d5aa8186f0 more doc mrq 2024-11-05 16:53:00 -0600
  • 9901c4f8ca documentation under ./docs/ mrq 2024-11-05 16:11:01 -0600
  • bbc2de3713 ugh mrq 2024-11-05 11:50:05 -0600
  • 9e65e05e83 more windows specific fixes, limit gradio to <5.0.0 on linux (it works on windows, but not on my linux machine tm) mrq 2024-11-04 18:00:33 -0600
  • c83670c38c Windows specific fixes (to-do: find libespeak-ng.dll automatically because it cannot be trusted to do it by default) mrq 2024-11-03 19:19:15 -0600
  • d229725c76 more adjustments (adjustments of early-exit entropy/varentropy thresholds, default rep pen being 1.5, experimental refine-on-stop, etc.) mrq 2024-11-03 18:31:28 -0600
  • aee08b7307 changed layerskip float16 training warning (since it didnt seem to fry on my 4xV100 system) mrq 2024-11-03 09:58:29 -0600
  • 3826f9bae4 saner mask creation? (it doesnt matter, kv cache wont work) mrq 2024-11-02 21:00:21 -0500
  • ded746e157 very, very naive layerskip speculative sampling (it just checks if the current layer's state is good enough) mrq 2024-11-02 11:49:05 -0500
  • 62fe5b0943 ughh mrq 2024-11-01 22:36:48 -0500
  • ec79230965 shuffled web UI options hidden by cfg.experimental to its own tab, expose early exit selection to inferencing (it kinda works naively, still need to implement self-speculation) mrq 2024-11-01 21:30:06 -0500
  • ef1c17430f skip step on nan loss (ironically I have not had a nan loss after adding this), throw exception with invalid cfg.dataset.sample_type and sample_order combination (because I was tricked by this in my yaml and had inconsistent vram usage) mrq 2024-11-01 20:54:53 -0500
  • fb8faa295b actually float16(+AMP) and layerskip is bad and will kill the model...... mrq 2024-11-01 18:36:44 -0500
  • edf1e66bf9 layerskip_r=6 fries the model so hard the loss is sub-1... mrq 2024-11-01 17:06:07 -0500
  • 9b6c57bc57 third time's the charm (for some reason it escaped me that I should treat early exit loss as an aux_loss to be used with the normal loss, as if I was training a MoE's router) mrq 2024-11-01 12:50:37 -0500
  • 76ebef45dc off-by-one... mrq 2024-10-31 13:24:48 -0500
  • b63293cbbe ugh mrq 2024-10-30 22:49:11 -0500
  • a22534e8f4 layer skip training implemented (need to gut the inferencing from the repo, and to actually see if the model can benefit from this) mrq 2024-10-30 20:05:45 -0500
  • 4049f51ba9 added option to load lora directly from the model file itself with --lora mrq 2024-10-26 00:13:10 -0500
  • 023c3af331 updated readme to reflect changes mrq 2024-10-25 22:17:05 -0500
  • ccf71dc1b6 added option to load from a model state dict directly instead of a yaml (to-do: do this for LoRAs too), automatically download the default model if none is provided mrq 2024-10-25 22:15:15 -0500
  • a96f5aee32 adjusted how i want to pass eval kwargs mrq 2024-10-25 20:38:09 -0500
  • 92e6bff6dc actually ar temp 0.5 with rep pen 1.125 seems to have the benefits of better outputs without it degrading some of the time but not all the time mrq 2024-10-23 00:03:35 -0500
  • 8920e5e86b actually have beam_width in the webUI work mrq 2024-10-22 22:06:22 -0500
  • 910571ad34 too brainlet to diagnose why low temp / greedy sampling is randomly unstable some of the time mrq 2024-10-22 20:13:54 -0500
  • 8eb9a4056b modified default arguments (ar temp = 0 and rep pen = 1.125 seems to be stable, at least given the few things i tested), do not pass top k/top p/min p to NAR even though technically none of those things should matter when greedy sampling mrq 2024-10-22 18:12:39 -0500
  • 1a02cd5bce modify demo template to say F5 instead of YourTTS, swap LoRA comparison around to make the lora'd the base file, and the no-lora the suffix'd file mrq 2024-10-21 19:52:02 -0500
  • 02dfc60ac3 ugh mrq 2024-10-18 17:23:22 -0500
  • 71731ed785 added prefixing with silence (was to test something, currently hidden under cfg.experimental=True) mrq 2024-10-18 17:19:52 -0500
  • 6b04c13c56 print warning if audio promtpless inferencing with low AR temp (it really doesn't like low temps / greedy sampling) mrq 2024-10-18 17:01:40 -0500
  • c8f31db1de default to greedy sample AR (i should probably test this more but it seems to pass my harvard sentences and tongue twisters) mrq 2024-10-18 16:58:56 -0500
  • fc8dfd8617 made greedy AR sampling viable (and preferable), with caveats (per comment in vall_e.models.ar_nar) mrq 2024-10-18 16:55:00 -0500
  • 07f4935a75 more tweaks mrq 2024-10-18 13:19:36 -0500
  • 0dfab973e7 oops mrq 2024-10-18 09:40:06 -0500
  • 75b90be325 cleaned up unused config flags, allow less strict yaml by pruning missing keys, renamed some dataset configs to be more unified mrq 2024-10-17 17:06:48 -0500
  • 8b6095f681 saner defaults, maybe mrq 2024-10-17 14:37:21 -0500
  • f88097ccf6 add config option to set the rate of sampling randomly vs similar speakers during training mrq 2024-10-16 14:27:58 -0500
  • 48461833c2 ugh mrq 2024-10-15 19:30:43 -0500
  • eea70f5698 kludge fix for an oversight in the model when trying to train for longer input prompt durations...... mrq 2024-10-15 19:25:03 -0500
  • 84005c5b00 entropix apparently processes the entire sequence of logits but it falls apart when doing that mrq 2024-10-13 12:01:12 -0500
  • c800d28bb8 respect attention defined in the yaml for web UI (which might explain why theres been a discrepancy in outputs for me) mrq 2024-10-13 11:02:24 -0500
  • ed6b7a690f ugh......... mrq 2024-10-13 00:26:46 -0500
  • d405f243d4 at wits end in trying to output the right attention scores mrq 2024-10-12 23:53:13 -0500
  • 70cf694cfd output attention scores for SDPA/flash, since naive attention seems broken mrq 2024-10-12 12:09:17 -0500
  • 541e45263c ugh mrq 2024-10-12 11:29:16 -0500
  • 04e983b86b modified demo page to be more modular with demoing comparisons, actually provide a path to use modified naive attention, entropix sampling is not tied to an experimental yaml flag now mrq 2024-10-12 11:27:55 -0500
  • 666e8038fb ugh mrq 2024-10-12 10:41:35 -0500
  • 3d6ef9666b overridden naive llama attention to get the right score values that entropix needs mrq 2024-10-12 10:05:47 -0500
  • 40b089daf3 lol mrq 2024-10-12 09:57:34 -0500
  • d6f7c86a5c entropix tweaks (it doesn't output garbage but it loves to go for silence) mrq 2024-10-12 09:46:18 -0500
  • d0ab7d755a added min-p (really does not seem useful since it's very sensitive), more tweaks to entropix mrq 2024-10-11 22:36:06 -0500
  • bef43a0c18 added experimental entropix sampling support mrq 2024-10-11 21:18:26 -0500
  • 85d85c1351 more arg creep for demo page mrq 2024-10-10 19:40:01 -0500
  • 301468f519 << mrq 2024-10-10 19:13:52 -0500
  • 75a4c866d6 more demo page tweaks, added arg to force enable/disable LoRAs for inferencing (to-do: setup arg flags to handle this, and checkbox in web UI) mrq 2024-10-10 19:04:12 -0500
  • 96d05be73c demo page tweaks mrq 2024-10-10 13:52:37 -0500
  • 2ea978f318 added --eval-random-text-prompts to use random text prompts for eval pass, added --random-prompts for demo page and --lora to use a sample with the lora disabled, probably finally fixed validation dataloader breaking on eval mrq 2024-10-10 13:40:25 -0500
  • 52299127ab fix vall_e.emb.process mrq 2024-10-08 20:00:34 -0500
  • 0656a762af fix vall_e.emb.transcriber mrq 2024-10-08 19:24:43 -0500
  • acdce66d4e readme tweaks, set the (unused) default model download URL back to the base ar+nar-llama-8 model, as ar+nar-tts+stt-llama-8 was renamed back to it since it performs well mrq 2024-10-05 22:53:53 -0500
  • 84c7419001 faster mrq 2024-10-04 22:30:47 -0500