System Info
root@12bb27e08b1b:/# pip show transformers
Name: transformers
Version: 4.52.3
trainer.py contains this:
if is_apex_available():
from apex import amp
Apex (built from source, as they recommend) does no longer come with amp.
How to reproduce?
- install transformers
- install apex
- python
from trl import SFTTrainer
Who can help?
No response
Information
Tasks
Reproduction
How to reproduce?
- install transformers
- install apex
- python
from trl import SFTTrainer
Expected behavior
There should not be from apex import amp in the code base