Merge pull request #169 from julian-parker/patch-3

Add correct safetensors import

Author: zqevans

stable_audio_tools/training/autoencoders.py

@@ -18,6 +18,7 @@
 from ..interface.aeiou import audio_spectrogram_image, tokens_spectrogram_image
 from ema_pytorch import EMA
 from einops import rearrange
+from safetensors.torch import save_model
 
 def trim_to_shortest(a, b):
     """Trim the longer of two tensors to the length of the shorter one."""