Made a very basic vae trainer.

2023-07-17 19:03:50 -06:00
parent 78b59c5e99
commit 439310e4dc
11 changed files with 410 additions and 89 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 torch
+torchvision
 safetensors
 diffusers
 transformers