Merge pull request #41 from abinitio888/main

Fix import partial
lucidrains · Mar 31, 2023 · 4d74aa0 · 4d74aa0
2 parents 80ad8cc + 64bae47
commit 4d74aa0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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