Spaces:
Runtime error
Runtime error
Hugo Flores Garcia
commited on
Commit
·
b87f1ce
1
Parent(s):
d56c9c9
breakpoint
Browse files- 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(
|
| 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)
|