Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Users should specify the torch_dtype they want, and if they don't it will be torch.float32.