Commit ·
be6322a
1
Parent(s): 178e836
Update example usage in README.md
Browse files
README.md
CHANGED
|
@@ -46,6 +46,10 @@ model = AutoModel.from_pretrained("rustemgareev/mdeberta-v3-base-lite")
|
|
| 46 |
text = "This is an example text in English."
|
| 47 |
inputs = tokenizer(text, return_tensors="pt")
|
| 48 |
outputs = model(**inputs)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
```
|
| 50 |
|
| 51 |
## Performance Evaluation
|
|
|
|
| 46 |
text = "This is an example text in English."
|
| 47 |
inputs = tokenizer(text, return_tensors="pt")
|
| 48 |
outputs = model(**inputs)
|
| 49 |
+
|
| 50 |
+
# Verify that the model outputs hidden states (embeddings)
|
| 51 |
+
print("Success! Output shape (batch_size, sequence_length, hidden_size):")
|
| 52 |
+
print(outputs.last_hidden_state.shape)
|
| 53 |
```
|
| 54 |
|
| 55 |
## Performance Evaluation
|