farzadab commited on
Commit
124e049
1 Parent(s): de8c22f

Upload model

Browse files
config.json CHANGED
@@ -1,10 +1,10 @@
1
  {
2
  "architectures": [
3
- "ResnetModel"
4
  ],
5
  "auto_map": {
6
- "AutoConfig": "configuration_resnet.ResnetConfig",
7
- "AutoModel": "modeling_resnet.ResnetModel"
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 ResnetConfig(PretrainedConfig):
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:6af6e14b216661883cf506e24dc3c3ac81c66d329159118467bb5f7b9f48667e
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 ResnetConfig
4
 
5
 
6
- class ResnetModel(PreTrainedModel):
7
- config_class = ResnetConfig
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)