JohnSnowLabs / spark-nlp

State of the Art Natural Language Processing
https://sparknlp.org/
Apache License 2.0
3.88k stars 711 forks source link

[SPARKNLP-1027] Fix issue with pretrained model #14413

Closed DevinTDHa closed 1 month ago

DevinTDHa commented 1 month ago

Description

This PR fixes a critical bug when using pretrained models with AutoGGUFModels.

The file protocol is included in the path while trying to read the model. This has been (temporarily) fixed by removing the protocol from the path string. The file should be local at this point, as it as copied to a temporary local location if it was remote.

How Has This Been Tested?

Tests passing with pretrained models.

Types of changes