No description provided.

You're running into a CUDA-related runtime error while trying to load a Hugging Face model with device_map='cuda', but your environment does not have an NVIDIA GPU or driver installed.
so i added a conditional DEVICE parameter to use gpu oif supported else use cpu

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment