targ_list=[r[...,l]forr,linzip(resps_list,quant_levels)]# ensures we only have 1 RVQ-bin (our target)
resps_list=[rifl==0elser[...,:l]forr,linzip(resps_list,quant_levels)]# r[..., 0] is technically correct, but only r[:, 0] gets passed through the embedding
"""
ifcfg.experimental:
proms_list=[rifl==0elsetrim(r,75*3)forr,linzip(proms_list,quant_levels)]# trim input prompt to 3 seconds