Update README.md
Browse files
README.md
CHANGED
|
@@ -31,7 +31,7 @@ To encode (and reconstruct) audio with SNAC in Python, use the following code:
|
|
| 31 |
import torch
|
| 32 |
from snac import SNAC
|
| 33 |
|
| 34 |
-
model = SNAC.from_pretrained("hubertsiuzdak/
|
| 35 |
audio = torch.randn(1, 1, 44100).cuda() # B, 1, T
|
| 36 |
|
| 37 |
with torch.inference_mode():
|
|
|
|
| 31 |
import torch
|
| 32 |
from snac import SNAC
|
| 33 |
|
| 34 |
+
model = SNAC.from_pretrained("hubertsiuzdak/snac_44khz").eval().cuda()
|
| 35 |
audio = torch.randn(1, 1, 44100).cuda() # B, 1, T
|
| 36 |
|
| 37 |
with torch.inference_mode():
|