chunked/nltk_chunking/_quantization/chunk_91.txt

You don't need to pass the device_map parameter for training because it'll automatically load your model on a GPU.