How to use codesage/codesage-small with Transformers:
# Load model directly from transformers import CodeSage model = CodeSage.from_pretrained("codesage/codesage-small", trust_remote_code=True, dtype="auto")
Hello!
Before, the snippet gave only token-level embeddings, which are not very useful in practice generally. I think the pooler output will be more useful.
· Sign up or log in to comment