Mirrors
/
riffusion-inference
mirror of https://github.com/riffusion/riffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Convert to RGB

This commit is contained in:
Jasper 2022-12-16 16:26:46 -06:00 committed by GitHub
parent 9ab8c0f105
commit 1c5caf77ac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
riffusion/audio.py
View File

@ -106,8 +106,10 @@ def image_from_spectrogram(
# Flip Y
image = image.transpose(Image.FLIP_TOP_BOTTOM)
return image
# Convert to RGB
image = image.convert("RGB")
return image
def spectrogram_from_waveform(
waveform: np.ndarray,