---
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: 1000
- Output size: 1000
- 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_1k")

# Forward pass
import torch
x = torch.randn(1, 1000)  # Example input
output = model(x)
print(output.shape) # shoudl be 1,1000
```