[Feature]: Pin torch>=2.x to use torch.compile

[mauryaland]

Problem statement

Get benefits from features of torch>=2.x

Solution

Pin torch version to >=2.x to use torch.compile on models

Additional Context

No response

[helpmefindaname]

Hi @mauryaland
can you ebalorate why pinning torch >=2.x is necessary for that?
I don't see anything preventing you from jsut installing torch 2.x and using torch.compile on any flair model (or embedding)