ViewModelInjectionFactory.java

public class ViewModelInjectionFactory<VM extends ViewModel> implements ViewModelProvider.Factory {

    private Lazy<VM> viewModel;

    @Inject
    public ViewModelInjectionFactory(Lazy<VM> viewModel) {
        this.viewModel = viewModel;
    }

    @Override
    @SuppressWarnings("unchecked cast")
    public <T extends ViewModel> T create(Class<T> modelClass) {
        return (T) viewModel.get();
    }
}