Skip to content

Commit 7ed8ac9

Browse files
committed
fix readme
1 parent 5141c19 commit 7ed8ac9

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,10 @@ import torch
1717
from vector_quantize_pytorch import VectorQuantize
1818

1919
vq = VectorQuantize(
20-
dim = 256,
21-
n_embed = 512, # size of the dictionary
22-
decay = 0.8, # the exponential moving average decay, lower means the dictionary will change faster
23-
commitment = 1. # the weight on the commitment loss
20+
dim = 256,
21+
n_embed = 512, # size of the dictionary
22+
decay = 0.8, # the exponential moving average decay, lower means the dictionary will change faster
23+
commitment = 1. # the weight on the commitment loss
2424
)
2525

2626
x = torch.randn(1, 1024, 256)

0 commit comments

Comments
 (0)