lucidrains / PaLM-rlhf-pytorch

Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM
MIT License
7.67k stars 668 forks source link

Do you need cuda for this? #30

Closed beew closed 1 year ago

beew commented 1 year ago

Hi, just wondering if it would work without cuda. Thanks.

lucidrains commented 1 year ago

yes to title, no to the body