847c35f7d5
This commit adds the trust_remote_code=True parameter when loading models and configurations in the embedding model conversion scripts. It also adds a cast to float for models that might use a data type that is not supported by python, for example bfloat16. The motivation for this is that some models may require custom code to be executed during loading, and setting trust_remote_code to True avoids getting prompted for confirmation. Future work will consolidate the embedding conversion scripts with the causal conversion scripts to avoid code duplication. But in the mean time it would be nice to have this fix in place. |
||
|---|---|---|
| .. | ||
| compare-embeddings-logits.sh | ||
| convert-model.sh | ||
| modelcard.template | ||
| run-converted-model.sh | ||
| run-original-model.py | ||