Update readme
This commit is contained in:
@@ -82,6 +82,7 @@ for model_class, tokenizer_class, pretrained_weights in MODELS:
|
||||
|
||||
# Encode text
|
||||
input_ids = torch.tensor([tokenizer.encode("Here is some text to encode")])
|
||||
with torch.no_grad():
|
||||
last_hidden_states = model(input_ids)[0] # Models outputs are now tuples
|
||||
|
||||
# Each architecture is provided with several class for fine-tuning on down-stream tasks, e.g.
|
||||
|
||||
Reference in New Issue
Block a user