Closed hunterqueb closed 1 month ago
Ran into issues running Adam-mini on a non-cuda device. I noticed hardcoded calls that sent tensors to cuda and replaced it with to(device) calls.
Thanks for the refinement!
Ran into issues running Adam-mini on a non-cuda device. I noticed hardcoded calls that sent tensors to cuda and replaced it with to(device) calls.