Commit
·
09566db
1
Parent(s):
554e383
update
Browse files- __pycache__/model.cpython-313.pyc +0 -0
- config.json/config.json +1 -1
- model.py +1 -1
- pytorch_model.bin/config.json +1 -1
- pytorch_model.bin/model.safetensors +1 -1
__pycache__/model.cpython-313.pyc
CHANGED
|
Binary files a/__pycache__/model.cpython-313.pyc and b/__pycache__/model.cpython-313.pyc differ
|
|
|
config.json/config.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"input_dim": 4,
|
| 3 |
-
"model_type": "
|
| 4 |
"output_dim": 16,
|
| 5 |
"transformers_version": "4.54.1"
|
| 6 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"input_dim": 4,
|
| 3 |
+
"model_type": "test-model",
|
| 4 |
"output_dim": 16,
|
| 5 |
"transformers_version": "4.54.1"
|
| 6 |
}
|
model.py
CHANGED
|
@@ -2,7 +2,7 @@ from transformers import PreTrainedModel, PretrainedConfig
|
|
| 2 |
from torch import nn
|
| 3 |
|
| 4 |
class TestConfig(PretrainedConfig):
|
| 5 |
-
model_type = "
|
| 6 |
|
| 7 |
def __init__(self, input_dim=4, output_dim=16, **kwargs):
|
| 8 |
super().__init__(**kwargs)
|
|
|
|
| 2 |
from torch import nn
|
| 3 |
|
| 4 |
class TestConfig(PretrainedConfig):
|
| 5 |
+
model_type = "test-model"
|
| 6 |
|
| 7 |
def __init__(self, input_dim=4, output_dim=16, **kwargs):
|
| 8 |
super().__init__(**kwargs)
|
pytorch_model.bin/config.json
CHANGED
|
@@ -3,7 +3,7 @@
|
|
| 3 |
"TestModel"
|
| 4 |
],
|
| 5 |
"input_dim": 4,
|
| 6 |
-
"model_type": "
|
| 7 |
"output_dim": 16,
|
| 8 |
"torch_dtype": "float32",
|
| 9 |
"transformers_version": "4.54.1"
|
|
|
|
| 3 |
"TestModel"
|
| 4 |
],
|
| 5 |
"input_dim": 4,
|
| 6 |
+
"model_type": "test-model",
|
| 7 |
"output_dim": 16,
|
| 8 |
"torch_dtype": "float32",
|
| 9 |
"transformers_version": "4.54.1"
|
pytorch_model.bin/model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 496
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:05a3301854d248873c8f006dd97769830f867377aaf0a58821d1e0330b3d2e0d
|
| 3 |
size 496
|