hf_load_model: Load a pre-trained AutoModel object from Hugging Face

View source: R/transformers.R

hf_load_modelR Documentation

Load a pre-trained AutoModel object from Hugging Face

Description

Load a pre-trained AutoModel object from Hugging Face

Usage

hf_load_model(model_id, ...)

Arguments

model_id

model_id The id of the model given in the url by https://huggingface.co/model_name.

...

sent to AutoModel.from_pretrained()

Value

a pre-trained model object

Examples

## Not run: 
model <- hf_load_model("distilbert-base-uncased")

## End(Not run)

farach/huggingfaceR documentation built on Feb. 4, 2023, 10:31 p.m.