Hugo Flores Garcia commited on
Commit
b87f1ce
1 Parent(s): d56c9c9

breakpoint

Browse files
Files changed (1) hide show
  1. vampnet/interface.py +1 -2
vampnet/interface.py CHANGED
@@ -65,7 +65,7 @@ class Interface(torch.nn.Module):
65
  ):
66
  super().__init__()
67
  assert codec_ckpt is not None, "must provide a codec checkpoint"
68
- self.codec = DAC.load(Path(codec_ckpt))
69
  self.codec.eval()
70
  self.codec.to(device)
71
 
@@ -316,7 +316,6 @@ class Interface(torch.nn.Module):
316
  mask=mask_chunk,
317
  **kwargs
318
  )
319
- breakpoint()
320
  fine_z.append(chunk)
321
 
322
  fine_z = torch.cat(fine_z, dim=-1)
 
65
  ):
66
  super().__init__()
67
  assert codec_ckpt is not None, "must provide a codec checkpoint"
68
+ self.codec = DAC.load(codec_ckpt)
69
  self.codec.eval()
70
  self.codec.to(device)
71
 
 
316
  mask=mask_chunk,
317
  **kwargs
318
  )
 
319
  fine_z.append(chunk)
320
 
321
  fine_z = torch.cat(fine_z, dim=-1)