It would be useful to have an additional `device` parameter in `init_df` function to load the model on CPU.
It would be useful to have an additional
deviceparameter ininit_dffunction to load the model on CPU.