Hey @zohaibjan - the model used for inpainting objects is user selectable by changing the model_path: Uminosachi/realisticVisionV20_v20-inpainting line in the config file located at src/inpainting_config.yaml. You can select a model that is fine tuned for your specific domain using this configuration option. The dataset-forge repository does not contain functionality for fine tuning a model.
Hey @zohaibjan - the model used for inpainting objects is user selectable by changing the
model_path: Uminosachi/realisticVisionV20_v20-inpaintingline in the config file located atsrc/inpainting_config.yaml. You can select a model that is fine tuned for your specific domain using this configuration option. The dataset-forge repository does not contain functionality for fine tuning a model.