Sushil Singh commited on
Commit ·
77f799f
1
Parent(s): 2cea1e3
updated readme
Browse files
README.md
CHANGED
|
@@ -11,6 +11,9 @@ tags: []
|
|
| 11 |
|
| 12 |
## Model Details
|
| 13 |
|
|
|
|
|
|
|
|
|
|
| 14 |
### Model Description
|
| 15 |
|
| 16 |
<!-- Provide a longer summary of what this model is. -->
|
|
@@ -35,11 +38,28 @@ This is the model card of a 🤗 transformers model that has been pushed on the
|
|
| 35 |
|
| 36 |
## Uses
|
| 37 |
|
| 38 |
-
|
| 39 |
|
| 40 |
### Direct Use
|
| 41 |
|
| 42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
|
| 44 |
[More Information Needed]
|
| 45 |
|
|
|
|
| 11 |
|
| 12 |
## Model Details
|
| 13 |
|
| 14 |
+
- Input size: 768
|
| 15 |
+
- Output size: 512
|
| 16 |
+
- Architecture: Single linear layer
|
| 17 |
### Model Description
|
| 18 |
|
| 19 |
<!-- Provide a longer summary of what this model is. -->
|
|
|
|
| 38 |
|
| 39 |
## Uses
|
| 40 |
|
| 41 |
+
This is a test model does not have any practical use.
|
| 42 |
|
| 43 |
### Direct Use
|
| 44 |
|
| 45 |
+
## Usage
|
| 46 |
+
Since this is a custom model, the code associated with the model needs to be restiered before it can be instantiated.
|
| 47 |
+
To start with clone the repo
|
| 48 |
+
git clone https://huggingface.co/sushilks/simple_linear_model
|
| 49 |
+
|
| 50 |
+
```python
|
| 51 |
+
from simple_linear_model.model import *
|
| 52 |
+
from transformers import AutoConfig, AutoModel
|
| 53 |
+
|
| 54 |
+
# Load the model
|
| 55 |
+
model = AutoModel.from_pretrained("sushilks/simple-linear-model")
|
| 56 |
+
|
| 57 |
+
# Forward pass
|
| 58 |
+
import torch
|
| 59 |
+
x = torch.randn(1, 768) # Example input
|
| 60 |
+
output = model(x)
|
| 61 |
+
print(output.shape) # shoudl be 1,512
|
| 62 |
+
```
|
| 63 |
|
| 64 |
[More Information Needed]
|
| 65 |
|