OOM Error when running data pipeline tasks to train accessibility and performance bug models

[jpangas]

The data pipeline tasks to train the accessibility and performance models exit with code 137.
Accessibility : https://community-tc.services.mozilla.com/tasks/enJ0-SovQAeil8OsO1RxhA
Performance: https://community-tc.services.mozilla.com/tasks/L0piBWkWRSuseY2eNDRjcg

This could be because of using the compute-smaller worker type for these individual tasks. During CI, the models are trained on Taskcluster using the compute-large worker type.