Hugging Face's logo

Spaces:
teticio
/
audio-diffusion
Running

App Files Community
1
teticio commited on
Commit
f8a240c
•
1 Parent(s): af8111a

add attribution

Browse files
Files changed (2) hide show
  1. config/ldm_autoencoder_kl.yaml +2 -0
  2. scripts/train_vae.py +2 -0
config/ldm_autoencoder_kl.yaml CHANGED
@@ -1,4 +1,6 @@
1
 
 
 
2
  model:
3
  base_learning_rate: 4.5e-6
4
  target: ldm.models.autoencoder.AutoencoderKL
 
1
 
2
+ # based on https://github.com/CompVis/stable-diffusion/blob/main/configs/autoencoder/autoencoder_kl_32x32x4.yaml
3
+
4
  model:
5
  base_learning_rate: 4.5e-6
6
  target: ldm.models.autoencoder.AutoencoderKL
scripts/train_vae.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  import os
2
  import argparse
3
 
 
1
+ # based on https://github.com/CompVis/stable-diffusion/blob/main/main.py
2
+
3
  import os
4
  import argparse
5