NikitaSrivatsan commited on
Commit
01eda25
1 Parent(s): 6a4daf6

Added local version of CLAP checkpoint

Browse files
audiocaptioner.py CHANGED
@@ -65,4 +65,4 @@ class AudioCaptioner(torch.nn.Module):
65
  amodel = 'HTSAT-base'
66
  )
67
  with contextlib.redirect_stdout(io.StringIO()):
68
- self.clap_model.load_ckpt(ckpt = '/graft1/datasets/kechen/clap_ckpt/music_audioset_epoch_15_esc_90.14.pt')
 
65
  amodel = 'HTSAT-base'
66
  )
67
  with contextlib.redirect_stdout(io.StringIO()):
68
+ self.clap_model.load_ckpt(ckpt = '/checkpoints/music_audioset_epoch_15_esc_90.14.pt')
checkpoints/music_audioset_epoch_15_esc_90.14.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fae3e9c087f2909c28a09dc31c8dfcdacbc42ba44c70e972b58c1bd1caf6dedd
3
+ size 2352471003
infer.py CHANGED
@@ -18,7 +18,6 @@ def infer(input_filename):
18
  print(f'Loaded from {checkpoint}')
19
  model.eval()
20
  # read in the wav file and precompute neighbors
21
- #dataset_path = '/graft1/datasets/kechen/audiostock-full'
22
  dataset_path = ''
23
  train_dataset = AudiostockDataset(
24
  dataset_path=dataset_path,
 
18
  print(f'Loaded from {checkpoint}')
19
  model.eval()
20
  # read in the wav file and precompute neighbors
 
21
  dataset_path = ''
22
  train_dataset = AudiostockDataset(
23
  dataset_path=dataset_path,