rustemgareev commited on
Commit
be6322a
·
1 Parent(s): 178e836

Update example usage in README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -0
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