Commit 638b07b2 authored by Phil Wang's avatar Phil Wang
Browse files

avoid a division

parent d0cdbb25

musiclm_pytorch/musiclm_pytorch.py

+1 −1
Original line number Diff line number Diff line
@@ -543,7 +543,7 @@ class MuLaN(nn.Module): 


        denominator = reduce(cosine_sim_exp, 'i j -> i', 'sum')



        contrastive_loss = -log(numerator / denominator)

        contrastive_loss = -log(numerator) + log(denominator)

        return contrastive_loss.mean()



# music lm

setup.py

+1 −1
Original line number Diff line number Diff line
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages 
setup(

  name = 'musiclm-pytorch',

  packages = find_packages(exclude=[]),

  version = '0.0.21',

  version = '0.0.22',

  license='MIT',

  description = 'MusicLM - AudioLM + Audio CLIP to text to music synthesis',

  author = 'Phil Wang',