File size: 764 Bytes
016466e 9c3fa5d 016466e 9c3fa5d 016466e 9c3fa5d 016466e 9c3fa5d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | ---
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
```
|