The model initialization methods (components/models/GPT.py for example) should accept arguments to training_args and model config.
The model initialization methods (components/models/GPT.py for example) should accept arguments to training_args and model config.