Sushil Singh commited on
Commit
77f799f
·
1 Parent(s): 2cea1e3

updated readme

Browse files
Files changed (1) hide show
  1. README.md +22 -2
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
- <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
 
40
  ### Direct Use
41
 
42
- <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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