Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Enable BetterTransformer with the [PreTrainedModel.to_bettertransformer] method:
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder")
model.to_bettertransformer()
TorchScript
TorchScript is an intermediate PyTorch model representation that can be run in production environments where performance is important.