Update rita_modeling.py
Browse files- rita_modeling.py +1 -33
rita_modeling.py
CHANGED
@@ -1,35 +1,3 @@
|
|
1 |
-
|
2 |
-
Hugging Face's logo Hugging Face
|
3 |
-
|
4 |
-
Models
|
5 |
-
Datasets
|
6 |
-
Spaces
|
7 |
-
Docs
|
8 |
-
Pricing
|
9 |
-
Please check your email address for a confirmation link
|
10 |
-
nz
|
11 |
-
/
|
12 |
-
RITA_s
|
13 |
-
PyTorch
|
14 |
-
Transformers
|
15 |
-
rita
|
16 |
-
Model card
|
17 |
-
Files and versions
|
18 |
-
Settings
|
19 |
-
RITA_s
|
20 |
-
/ rita_modeling.py
|
21 |
-
nz's picture
|
22 |
-
nz
|
23 |
-
Update rita_modeling.py
|
24 |
-
b98e56e
|
25 |
-
about 21 hours ago
|
26 |
-
raw
|
27 |
-
history
|
28 |
-
blame
|
29 |
-
edit
|
30 |
-
delete
|
31 |
-
Safe
|
32 |
-
8.68 kB
|
33 |
import math
|
34 |
import os
|
35 |
from dataclasses import dataclass
|
@@ -278,4 +246,4 @@ class RITAModel(PreTrainedModel):
|
|
278 |
return self.projector
|
279 |
|
280 |
def set_output_embeddings(self, new_projector):
|
281 |
-
self.projector = new_projector
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import math
|
2 |
import os
|
3 |
from dataclasses import dataclass
|
|
|
246 |
return self.projector
|
247 |
|
248 |
def set_output_embeddings(self, new_projector):
|
249 |
+
self.projector = new_projector
|