Can dolly support torch.jit.trace?

[JackWeiw]

Environment:
transformer 4.40.1
torch 2.0.1+cu118
tokenizers 0.19.1

when i try torch.jit.trace the dolly-v2-3b model, it reports below, it smees dolly dosen't support torch.jit.trace, can anyone give me some isights?

![1714458130252](https://github.com/databrickslabs/dolly/assets/126441921/5b3df095-6e60-42ac-9544-267c

4eec0e81)