Commit
•
c69924f
1
Parent(s):
0fdbcd5
Upload model
Browse files- embedding_model.py +0 -2
embedding_model.py
CHANGED
|
@@ -2,11 +2,9 @@ import torch
|
|
| 2 |
import torch.nn.functional as F
|
| 3 |
from torch import nn
|
| 4 |
from transformers import BloomForCausalLM, PreTrainedModel
|
| 5 |
-
from .configuration import BufferEmbeddingConfig
|
| 6 |
|
| 7 |
|
| 8 |
class DualModel(PreTrainedModel):
|
| 9 |
-
config_class = BufferEmbeddingConfig
|
| 10 |
_auto_class = "AutoModel"
|
| 11 |
def __init__(self, config):
|
| 12 |
super(DualModel, self).__init__(config)
|
|
|
|
| 2 |
import torch.nn.functional as F
|
| 3 |
from torch import nn
|
| 4 |
from transformers import BloomForCausalLM, PreTrainedModel
|
|
|
|
| 5 |
|
| 6 |
|
| 7 |
class DualModel(PreTrainedModel):
|
|
|
|
| 8 |
_auto_class = "AutoModel"
|
| 9 |
def __init__(self, config):
|
| 10 |
super(DualModel, self).__init__(config)
|