FoleyCrafter

Runtime error

unknown commited on Jun 27, 2024

Commit

81fb07b

1 Parent(s): d6e439c

'torch_utils.py'

Files changed (1) hide show

foleycrafter/models/specvqgan/onset_baseline/utils/torch_utils.py CHANGED Viewed

@@ -10,7 +10,10 @@ import torch.nn.functional as F
 from torch.utils.data import DataLoader
 sys.path.append('..')
-import data
 # ---------------------------------------------------- #

 from torch.utils.data import DataLoader
 sys.path.append('..')
+try:
+    import data
+except:
+    import foleycrafter.models.specvqgan.onset_baseline.data
 # ---------------------------------------------------- #