src/model.py (1 line):
	- line 21: hidden_dim = md_dim # TODO (low priority): allow user to customize this