ColossalAI/applications/ChatGPT/chatgpt/models/base
wenjunyang b51bfec357
[chatgpt] change critic input as state (#3042)
* fix Critic

* fix Critic

* fix Critic

* fix neglect of attention mask

* fix neglect of attention mask

* fix neglect of attention mask

* add return

---------

Co-authored-by: yangwenjun <yangwenjun@soyoung.com>
Co-authored-by: yangwjd <yangwjd@chanjet.com>
2023-03-08 15:18:02 +08:00
..
__init__.py change nn to models (#3032) 2023-03-07 16:34:22 +08:00
actor.py change nn to models (#3032) 2023-03-07 16:34:22 +08:00
critic.py [chatgpt] change critic input as state (#3042) 2023-03-08 15:18:02 +08:00
reward_model.py change nn to models (#3032) 2023-03-07 16:34:22 +08:00