Fix device compatibility issue in `KAN.initialize_from_another_model`

[Jim137]

In the KAN.train function, the dataset is sent to the device within the function. However, for KAN.initialize_from_another_model, the dataset needs to be sent before calling the function. This PR allows the dataset to be sent to another_model.device, ensuring compatibility and consistency.