Skip to content

Commit 5215426

Browse files
authored
Fix 'ceil' is not defined
1 parent 7de0d36 commit 5215426

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

vector_quantize_pytorch/residual_fsq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import random
2-
from math import log2
2+
from math import ceil
33
from functools import partial
44

55
from typing import List

vector_quantize_pytorch/residual_lfq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import random
2-
from math import log2
2+
from math import ceil, log2
33
from functools import partial, cache
44

55
import torch

0 commit comments

Comments
 (0)