The UK government is set to consult on a scheme that allows AI firms to use existing musical works to train their models.