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.71k stars 669 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