vinothkannans commited on
Commit
d990db2
·
verified ·
1 Parent(s): 2a3815c

Initial commit

Browse files
Files changed (6) hide show
  1. .gitignore +1 -0
  2. LICENSE +21 -0
  3. README.md +3 -3
  4. config.json +8 -0
  5. index.py +26 -0
  6. model.safetensors +3 -0
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ __pycache__
LICENSE ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ MIT License
2
+
3
+ Copyright (c) 2025 Vinoth Kannan
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
README.md CHANGED
@@ -1,3 +1,3 @@
1
- ---
2
- license: mit
3
- ---
 
1
+ # Hello World AI Model
2
+
3
+ [![HuggingFace](https://img.shields.io/badge/huggingface-%23FFD21E.svg?style=for-the-badge&logo=huggingface&logoColor=white)](https://huggingface.co/vinothkannans/hello-world)
config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "HelloWorldModel"
4
+ ],
5
+ "dtype": "float32",
6
+ "model_type": "hello-world",
7
+ "transformers_version": "4.57.1"
8
+ }
index.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import PreTrainedModel, PretrainedConfig
2
+ import torch.nn as nn
3
+ import torch
4
+
5
+ class HelloWorldConfig(PretrainedConfig):
6
+ model_type = "hello-world"
7
+
8
+ class HelloWorldModel(PreTrainedModel):
9
+ config_class = HelloWorldConfig
10
+
11
+ def __init__(self, config):
12
+ super().__init__(config)
13
+ # A single linear layer just for demo
14
+ self.layer = nn.Linear(1, 1)
15
+
16
+ def forward(self, input_ids=None):
17
+ # Always returns “Hello World”
18
+ return {"text": "Hello World"}
19
+
20
+ # Create the model + config
21
+ config = HelloWorldConfig()
22
+ model = HelloWorldModel(config)
23
+
24
+ # Save to local folder
25
+ model.save_pretrained("./")
26
+ config.save_pretrained("./")
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:516bfc4da3e30dd6fd1e5be0b1faa66d13ca1ee082b20cc2f759e4435121d0ec
3
+ size 176