MyComposable.kt

val creationCallback: (MyViewModel.Factory) -> MyViewModel = { factory ->
  factory.create(runtimeArg = "abc")
}
val viewModel = viewModel<MyViewModel>(
  extras = requireNotNull(LocalViewModelStoreOwner.current).run {
    if (this is HasDefaultViewModelProviderFactory) {
      this.defaultViewModelCreationExtras.withCreationCallback(creationCallback)
    } else {
      CreationExtras.Empty.withCreationCallback(creationCallback)
    }
  }
)