Bugs: Custom Models
cc: @aying-gl
epic