Fix import partial

lucidrains · Mar 31, 2023 · 64bae47 · 64bae47
1 parent 80ad8cc
commit 64bae47
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