File size: 689 Bytes
c1a41d7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
from . import latticed4, latticee8_padded12, half_integer_4bit_1col
# name: (id, codebook class)
codebook_id = {
'D4': (0, latticed4.D4_codebook),
'E8P12': (7, latticee8_padded12.E8P12_codebook),
'HI4B1C': (10, half_integer_4bit_1col.HI4B1C_codebook),
}
# id from above:6quantized linear implementation
quantized_class = {
0: latticed4.QuantizedD4Linear,
7: latticee8_padded12.QuantizedE8P12Linear,
10: half_integer_4bit_1col.QuantizedHI4B1CLinear,
}
cache_permute_set = {
0, # D4
}
def get_codebook(name):
return codebook_id[name][1]()
def get_id(name):
return codebook_id[name][0]
def get_quantized_class(id):
return quantized_class[id]
|