Trained on small dataset, model still in progress and there is a lot to improve to be honest. Trained on photogasm sd model.