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