ColossalAI/applications/ChatGPT/chatgpt
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
..
dataset [chatgpt] fix rm eval (#2829) 2023-02-21 11:35:45 +08:00
experience_maker change nn to models (#3032) 2023-03-07 16:34:22 +08:00
models [chatgpt] change critic input as state (#3042) 2023-03-08 15:18:02 +08:00
replay_buffer [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00
trainer change nn to models (#3032) 2023-03-07 16:34:22 +08:00
__init__.py [app] add chatgpt application (#2698) 2023-02-14 22:17:25 +08:00