codebook_init: guard torch.library.define against re-registration (multi-ckpt in one kernel)

trust_remote_code: codebook_init.py