src/transformers/integrations/integration_utils.py · 1c7ebf1d6eaf0ed0fd4101fd6eb7e64601429cfe · chenpangpang / transformers

don't log base model architecture in wandb if log model is false (#32143) · 1c7ebf1d

João Nadkarni authored Jul 26, 2024



* don't log base model architecture in wandb is log model is false

* Update src/transformers/integrations/integration_utils.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* convert log model setting into an enum

* fix formatting

---------
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

1c7ebf1d

integration_utils.py 92.7 KB

Replace integration_utils.py