Commit Graph

192 Commits

Author SHA1 Message Date
James Betker
2158383fa4 Revert previous changes 2022-06-10 14:34:05 -06:00
James Betker
89bd40d39f eval bug fix 2022-06-10 13:51:06 -06:00
James Betker
84469f3538 get rid of encoder checkpointing 2022-06-10 10:50:34 -06:00
James Betker
97b32dd39d try to make tfd8 be able to be trained e2e in quantizer mode 2022-06-10 10:40:56 -06:00
James Betker
e78c4b422c tfd8 2022-06-10 09:24:41 -06:00
James Betker
d98b895307 loss aware fix and report gumbel temperature 2022-06-09 21:56:47 -06:00
James Betker
47b34f5cb9 mup work checkin 2022-06-09 21:15:09 -06:00
James Betker
e67e82be2d misc 2022-06-09 21:14:48 -06:00
James Betker
16936881e5 allow freezing the upper quantizer 2022-06-08 18:30:22 -06:00
James Betker
43f225c35c debug gumbel temperature 2022-06-08 12:12:08 -06:00
James Betker
91be38cba3 . 2022-06-08 11:54:46 -06:00
James Betker
dee2b72786 checkpointing bugs, smh 2022-06-08 11:53:10 -06:00
James Betker
c61cd64bc9 network updates 2022-06-08 09:26:59 -06:00
James Betker
5a54d7db11 unet with ar prior 2022-06-07 17:52:36 -06:00
James Betker
5028703b3d ci not required 2022-06-06 09:26:25 -06:00
James Betker
08597bfaf5 fix 2022-06-06 09:21:58 -06:00
James Betker
49568ee16f some updates 2022-06-06 09:13:47 -06:00
James Betker
602df0abbc revert changes to dietattentionblock 2022-06-05 10:06:17 -06:00
James Betker
51d1908e94 update 2022-06-05 09:35:43 -06:00
James Betker
f9ebcf11d8 fix2 2022-06-05 01:31:37 -06:00
James Betker
aac92b01b3 fix 2022-06-05 01:27:28 -06:00
James Betker
38d8b17d18 tfd8 gets real verbose grad norm metrics 2022-06-04 23:09:54 -06:00
James Betker
0a9d4d4afc bunch of new stuff 2022-06-04 22:23:08 -06:00
James Betker
8f8b189025 Support legacy vqvae quantizer in music_quantizer 2022-06-04 10:16:24 -06:00
James Betker
40ba802104 padding 2022-06-03 12:09:59 -06:00
James Betker
581bc7ac5c udmc update 2022-06-03 12:02:22 -06:00
James Betker
9d8c2bddb1 classical unet for music 2022-06-03 11:03:14 -06:00
James Betker
2f4d990ad1 tfd7 2022-06-02 09:27:40 -06:00
James Betker
b2a83efe50 a few fixes 2022-06-01 16:35:15 -06:00
James Betker
712e0e82f7 fix bug 2022-06-01 14:21:44 -06:00
James Betker
de54be5570 propagate diversity loss 2022-06-01 14:18:50 -06:00
James Betker
4c6ef42b38 freeze quantizer until step 2022-06-01 08:06:05 -06:00
James Betker
64b6ae2f4a fix 2022-06-01 01:01:32 -06:00
James Betker
1ac02acdc3 tfd7 2022-06-01 00:50:40 -06:00
James Betker
e8cb93a4e9 fix size issues 2022-05-31 21:23:26 -06:00
James Betker
8a1b8e3e62 add checkpointing 2022-05-31 21:09:05 -06:00
James Betker
c0db85bf4f music quantizer 2022-05-31 21:06:54 -06:00
James Betker
29b55d42a5 one more 2022-05-30 16:33:49 -06:00
James Betker
71cf654957 fix unused parameters 2022-05-30 16:31:40 -06:00
James Betker
f7d237a50a train quantizer with diffusion 2022-05-30 16:25:33 -06:00
James Betker
136021bf8d tfd6 2022-05-30 09:09:42 -06:00
James Betker
536c8558ae fix 2022-05-28 22:32:38 -06:00
James Betker
da367da411 df5 2022-05-28 22:30:23 -06:00
James Betker
86694aef4e tfd5 2022-05-28 22:27:04 -06:00
James Betker
b6b4f10e1b ... 2022-05-28 10:59:03 -06:00
James Betker
0d3b831cf9 big fatty 2022-05-28 10:55:43 -06:00
James Betker
490d39b967 some stuff 2022-05-27 11:40:31 -06:00
James Betker
0659fe3d1e tfd3 mods 2022-05-27 11:16:26 -06:00
James Betker
c46da0285c Move stuff around 2022-05-27 11:06:58 -06:00
James Betker
9852599b34 tfd5 - with clvp! 2022-05-27 09:49:10 -06:00
James Betker
3db862dd32 adf update 2022-05-27 09:25:53 -06:00
James Betker
8587a18717 fd fix 2022-05-26 20:19:09 -06:00
James Betker
dd13b883ac td4 2022-05-26 14:56:03 -06:00
James Betker
1dbe0b6b2e a 2022-05-26 10:13:27 -06:00
James Betker
aa653115f1 tfd3 2022-05-26 10:09:11 -06:00
James Betker
36c68692a6 forgot to add rotary embeddings 2022-05-26 09:25:42 -06:00
James Betker
8ce48f04ff transformer diffusion 2 2022-05-26 09:08:35 -06:00
James Betker
52a20f3aa3 und10 2022-05-25 12:19:21 -06:00
James Betker
8b4b5ffa72 slight rework 2022-05-24 14:38:37 -06:00
James Betker
65b441d74e transformer diffusion 2022-05-24 14:02:05 -06:00
James Betker
4093e38717 revert flat diffusion back... 2022-05-22 23:10:58 -06:00
James Betker
1095248caf Revert "retest"
This reverts commit ed7768c73b.
2022-05-22 19:23:01 -06:00
James Betker
ed7768c73b retest 2022-05-22 16:30:09 -06:00
James Betker
2dd0b9e6e9 mel_head should be optional 2022-05-22 12:25:45 -06:00
James Betker
0c60f22197 fix unused parameters 2022-05-22 08:16:31 -06:00
James Betker
57d6f6d366 Big rework of flat_diffusion
Back to the drawing board, boys. Time to waste some resources catching bugs....
2022-05-22 08:09:33 -06:00
James Betker
be937d202e new attempt 2022-05-20 17:04:22 -06:00
James Betker
968660c248 another update 2022-05-20 11:25:00 -06:00
James Betker
28f950b7d3 fix 2022-05-20 11:18:52 -06:00
James Betker
b317c68ac9 fix 2022-05-20 11:12:53 -06:00
James Betker
3121bc4e43 flat diffusion 2022-05-20 11:01:48 -06:00
James Betker
e9fb2ead9a m2v stuff 2022-05-20 11:01:17 -06:00
James Betker
7812c23c7a revert fill_gaps back to old masking behavior 2022-05-08 00:10:19 -06:00
James Betker
58ed27d7a8 new gap_filler 2022-05-07 12:44:23 -06:00
James Betker
6c8032b4be more work 2022-05-06 21:56:49 -06:00
James Betker
79543e5488 Simpler form of the wavegen model 2022-05-06 16:37:04 -06:00
James Betker
d8925ccde5 few things with gap filling 2022-05-06 14:33:44 -06:00
James Betker
b13d983c24 and mel_head 2022-05-06 00:25:27 -06:00
James Betker
d5fb79564a remove mel_pred 2022-05-06 00:24:05 -06:00
James Betker
e9bb692490 fixed aligned_latent 2022-05-06 00:20:21 -06:00
James Betker
1609101a42 musical gap filler 2022-05-05 16:47:08 -06:00
James Betker
d66ab2d28c Remove unused waveform_gens 2022-05-04 21:06:54 -06:00
James Betker
ab219fbefb output variance 2022-05-02 00:10:33 -06:00
James Betker
3b074aac34 add checkpointing 2022-05-02 00:07:42 -06:00
James Betker
ae5f934ea1 diffwave 2022-05-02 00:05:04 -06:00
James Betker
afa2df57c9 gen3 2022-04-30 10:41:38 -06:00
James Betker
8aa6651fc7 fix surrogate loss return in waveform_gen2 2022-04-28 10:10:11 -06:00
James Betker
9df85c902e New gen2
Which is basically a autoencoder with a giant diffusion appendage attached
2022-04-20 21:37:34 -06:00
James Betker
24fdafd855 fix2 2022-04-20 00:03:29 -06:00
James Betker
0af0051399 fix 2022-04-20 00:01:57 -06:00
James Betker
419f4d37bd gen2 music 2022-04-19 23:38:37 -06:00
James Betker
546ecd5aeb music! 2022-04-15 21:21:37 -06:00