training/train_amused.py (1 line):
	- line 452: # TODO - will have to fix loading if training text encoder