unknown commited on
Commit
81fb07b
1 Parent(s): d6e439c

'torch_utils.py'

Browse files
foleycrafter/models/specvqgan/onset_baseline/utils/torch_utils.py CHANGED
@@ -10,7 +10,10 @@ import torch.nn.functional as F
10
  from torch.utils.data import DataLoader
11
 
12
  sys.path.append('..')
13
- import data
 
 
 
14
 
15
 
16
  # ---------------------------------------------------- #
 
10
  from torch.utils.data import DataLoader
11
 
12
  sys.path.append('..')
13
+ try:
14
+ import data
15
+ except:
16
+ import foleycrafter.models.specvqgan.onset_baseline.data
17
 
18
 
19
  # ---------------------------------------------------- #