sushilks/simple_linear_model_15k

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: 15000
• Output size: 15000
• Architecture: Single linear layer

Usage

To start with clone the repo git clone https://huggingface.co/sushilks/simple_linear_model

from simple_linear_model.model import *
from transformers import AutoConfig, AutoModel

from transformers import AutoModel
model = AutoModel.from_pretrained("sushilks/simple_linear_model_15k")

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