Add two new keyword parameters to set module tensor to device in src/accelerate/utils/modeling.py :

Add two new keyword parameters to set module tensor to device in src/accelerate/utils/modeling.py :: a task in HF ML Tasksmith (Harbor dataset). Both parameters must be accepted without error. The tensor must end up on the correct device with the correct dtype regardless of which combination of…