Spaces:
Running
on
T4
Running
on
T4
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)
|