New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Added patch for apple silicon #83

Open

mohit-s96 wants to merge 1 commit into serp-ai:main from mohit-s96:patch-apple-silicon

mohit-s96 commented Nov 3, 2023

This fixes the

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

error on Apple silicon chips when loading the tokenizer here:

tokenizer = CustomTokenizer.load_from_checkpoint('data/models/hubert/tokenizer.pth').to(device)

After the update we can use

tokenizer = CustomTokenizer.load_from_checkpoint('data/models/hubert/tokenizer.pth', map_location=torch.device(device))


          added patch for apple silicon chips

9f190d8

dagshub bot commented Nov 3, 2023

Join the discussion on DagsHub!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet