【踩坑日记9】TypeError: Accelerator.__init__() got an unexpected keyword argument ‘logging_dir‘

问题描述1

Traceback (most recent call last):
  File "/mnt/lab/XXX/code/promptist/./diff_prompter/ppo_prompter.py", line 276, in <module>
    main(args)
  File "/mnt/lab/XXX/code/promptist/./diff_prompter/ppo_prompter.py", line 264, in main
    model = trlx.train(
  File "/mnt/lab/XXX/anaconda3/envs/promptist/lib/python3.9/site-packages/trlx/trlx.py", line 47, in train
    model: AcceleratePPOModel = get_model(config.model.model_type)(config)
  File "/mnt/lab/XXX/anaconda3/envs/promptist/lib/python3.9/site-packages/trlx/model/accelerate_ppo_model.py", line 38, in __init__
    super().__init__(config)
  File "/mnt/lab/XXX/anaconda3/envs/promptist/lib/python3.9/site-packages/trlx/model/accelerate_base_model.py", line 31, in __init__
    self.accelerator = Accelerator(log_with="wandb", logging_dir=config.train.logging_dir)
TypeError: __init__() got an unexpected keyword argument 'logging_dir'

问题分析

在 accelerate 0.18.x版本之后logging_dir这个参数已经被弃用并且被删除，现在采用的是 project_dir 参数。

解决方法

重新安装accelerate包，降级为0.18.0版本

pip install accelerate==0.18.0

问题描述2

(promptist) longyy@sf:~/code/promptist/trlx$ pip install accelerate==0.18.0
Collecting accelerate==0.18.0
Downloading accelerate-0.18.0-py3-none-any.whl (215 kB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 215.3/215.3 kB 895.3 kB/s eta 0:00:00
Installing collected packages: accelerate
  Attempting uninstall: accelerate
    Found existing installation: accelerate 0.28.0
    Uninstalling accelerate-0.28.0:
      Successfully uninstalled accelerate-0.28.0
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
peft 0.10.1.dev0 requires accelerate>=0.21.0, but you have accelerate 0.18.0 which is incompatible.
Successfully installed accelerate-0.18.0

问题解决

先忽视

参考

【BUG】TypeError: Accelerator.init() got an unexpected keyword argument ‘logging_dir‘