Fix torch fx for LLama model

• Fix for Neuron (#30259)
• Fix copies for DBRX - neuron fix (#30610)

Thanks @michaelbenayoun !