| language: en | |
| license: mit | |
| tags: | |
| - simple_linear_model | |
| library_name: transformers | |
| # Custom Linear Model | |
| This is a simple custom model with a single linear layer. | |
| It does not have practical usages, just used for testing. | |
| ## Model Details | |
| - Input size: 30000 | |
| - Output size: 30000 | |
| - Architecture: Single linear layer | |
| ## Usage | |
| To start with clone the repo | |
| git clone https://huggingface.co/sushilks/simple_linear_model | |
| ```python | |
| from simple_linear_model.model import * | |
| from transformers import AutoConfig, AutoModel | |
| from transformers import AutoModel | |
| model = AutoModel.from_pretrained("sushilks/simple_linear_model_30k") | |
| # Forward pass | |
| import torch | |
| x = torch.randn(1, 30000) # Example input | |
| output = model(x) | |
| print(output.shape) # shoudl be 1,30000 | |
| ``` | |