Commit
•
82fdcc2
1
Parent(s):
91b2861
Upload model
Browse files- model.py +2 -0
- pytorch_model.bin +1 -1
model.py
CHANGED
|
@@ -12,10 +12,12 @@ from in2in.models.in2in import in2IN
|
|
| 12 |
from .config import in2INConfig
|
| 13 |
|
| 14 |
class in2INModel(PreTrainedModel):
|
|
|
|
| 15 |
config_class = in2INConfig
|
| 16 |
|
| 17 |
def __init__(self, config):
|
| 18 |
super().__init__(config)
|
|
|
|
| 19 |
self.model = in2IN(config, mode=config.MODE)
|
| 20 |
|
| 21 |
def forward(self, prompt_interaction, prompt_individual1, prompt_individual2):
|
|
|
|
| 12 |
from .config import in2INConfig
|
| 13 |
|
| 14 |
class in2INModel(PreTrainedModel):
|
| 15 |
+
|
| 16 |
config_class = in2INConfig
|
| 17 |
|
| 18 |
def __init__(self, config):
|
| 19 |
super().__init__(config)
|
| 20 |
+
self.mode = config.MODE
|
| 21 |
self.model = in2IN(config, mode=config.MODE)
|
| 22 |
|
| 23 |
def forward(self, prompt_interaction, prompt_individual1, prompt_individual2):
|
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1242367342
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b6a283bb5b46e53f95f2283c0a1825e9467ca0cc4d4e29c7a153a783df1f2cbf
|
| 3 |
size 1242367342
|