dnagpt
/

human_gpt2-v1

Feature Extraction

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

marisming commited on Aug 6, 2023

Commit

4c82835

·

1 Parent(s): 95077c5

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -38,4 +38,6 @@ print(embedding_mean.shape) # expect to be 768
 # embedding with max pooling
 embedding_max = torch.max(hidden_states[0], dim=0)[0]
-print(embedding_max.shape) # expect to be 768

 # embedding with max pooling
 embedding_max = torch.max(hidden_states[0], dim=0)[0]
+print(embedding_max.shape) # expect to be 768