Rahul Vinod Vishwakarma

icon indicating an email [email protected]

icon location Learning ML, DL, CNN, NLP.

Results 1 comments of Rahul Vinod Vishwakarma

LayoutLM.from_pretrained doesn't load embeddings' weights when using safetensors

In [SFconvertbot](https://huggingface.co/SFconvertbot)'s `convert.py` file, the loading of weights happens with - ``` loaded = torch.load(pt_filename, map_location="cpu", weights_only=True) ``` , which does not maps the layers correctly (the 'keys' in the...