From 64bae470052f285898aa356b5947c431e4882757 Mon Sep 17 00:00:00 2001
From: CJ <sejinc@tetrapak.com>
Date: Fri, 31 Mar 2023 15:15:07 +0200
Subject: [PATCH] Fix import partial

---
 musiclm_pytorch/musiclm_pytorch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/musiclm_pytorch/musiclm_pytorch.py b/musiclm_pytorch/musiclm_pytorch.py
index d7c9af7..58702f1 100644
--- a/musiclm_pytorch/musiclm_pytorch.py
+++ b/musiclm_pytorch/musiclm_pytorch.py
@@ -1,5 +1,5 @@
 import math
-from functools import wraps
+from functools import wraps, partial
 
 import torch
 import torch.nn.functional as F