Update model config to fix architecture mismatch

#159

This PR fixes an architecture mismatch in the model config. Please review carefully.

This comment has been hidden
Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment