Added more flexibility with UI components for model pruning & resuming to train.
--- support for the diffusers code-base is next
Added more flexibility with UI components for model pruning & resuming to train.
--- support for the diffusers code-base is next