Team,
its getting ltos of difficult to setup the environemtn with the right set of the library versions. literally spending lots of efforts to resolve the conflicts between torch, cuda, flash attn and other library. Its better to maintain specific versions for each library.
Thanks,