davidmrau/mixture-of-experts

Why not gpu?

Closed this issue · 1 comments

README says the code is on cpu, but i want to know whether the code can run on GPU, such as A100?

sure, it can be run on GPU, i will update the code