Joe Nguyen

Thank you! And also `tokens` in ``` self.token_cache[f"{string}_{max_len}"] = { k: v.squeeze(0).cpu().numpy() for k, v in tokens } ``` I assume it should be `tokens.items()`?

Also is there any possibility that you can publish the trained weight? Thank you for your help!