Upload model
Browse files- config.json +3 -3
- configuration_resnet.py +1 -1
- model.safetensors +1 -1
- modeling_resnet.py +3 -3
config.json
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
{
|
2 |
"architectures": [
|
3 |
-
"
|
4 |
],
|
5 |
"auto_map": {
|
6 |
-
"AutoConfig": "configuration_resnet.
|
7 |
-
"AutoModel": "modeling_resnet.
|
8 |
},
|
9 |
"avg_down": false,
|
10 |
"base_width": 64,
|
|
|
1 |
{
|
2 |
"architectures": [
|
3 |
+
"ResinModel"
|
4 |
],
|
5 |
"auto_map": {
|
6 |
+
"AutoConfig": "configuration_resnet.ResinConfig",
|
7 |
+
"AutoModel": "modeling_resnet.ResinModel"
|
8 |
},
|
9 |
"avg_down": false,
|
10 |
"base_width": 64,
|
configuration_resnet.py
CHANGED
@@ -2,7 +2,7 @@ from transformers import PretrainedConfig
|
|
2 |
from typing import List
|
3 |
|
4 |
|
5 |
-
class
|
6 |
model_type = "resin"
|
7 |
|
8 |
def __init__(
|
|
|
2 |
from typing import List
|
3 |
|
4 |
|
5 |
+
class ResinConfig(PretrainedConfig):
|
6 |
model_type = "resin"
|
7 |
|
8 |
def __init__(
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 416
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:532f1de9ef426ac086bb7d8d6648ded84230c3b628f034066984d1c2b08ed271
|
3 |
size 416
|
modeling_resnet.py
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
import torch
|
2 |
from transformers import PreTrainedModel
|
3 |
-
from .configuration_resnet import
|
4 |
|
5 |
|
6 |
-
class
|
7 |
-
config_class =
|
8 |
|
9 |
def __init__(self, config):
|
10 |
super().__init__(config)
|
|
|
1 |
import torch
|
2 |
from transformers import PreTrainedModel
|
3 |
+
from .configuration_resnet import ResinConfig
|
4 |
|
5 |
|
6 |
+
class ResinModel(PreTrainedModel):
|
7 |
+
config_class = ResinConfig
|
8 |
|
9 |
def __init__(self, config):
|
10 |
super().__init__(config)
|