Update custom_heads_flash_t5.py
Browse files- custom_heads_flash_t5.py +1 -45
custom_heads_flash_t5.py
CHANGED
|
@@ -1,47 +1,3 @@
|
|
| 1 |
-
|
| 2 |
-
Hugging Face's logo Hugging Face
|
| 3 |
-
|
| 4 |
-
Models
|
| 5 |
-
Datasets
|
| 6 |
-
Spaces
|
| 7 |
-
Posts
|
| 8 |
-
Docs
|
| 9 |
-
Pricing
|
| 10 |
-
|
| 11 |
-
CATIE-AQ
|
| 12 |
-
/
|
| 13 |
-
FAT5-small-UL2-fr
|
| 14 |
-
private
|
| 15 |
-
Feature Extraction
|
| 16 |
-
Transformers
|
| 17 |
-
Safetensors
|
| 18 |
-
French
|
| 19 |
-
flash_t5
|
| 20 |
-
flash-attention
|
| 21 |
-
UL2
|
| 22 |
-
FAT5
|
| 23 |
-
custom_code
|
| 24 |
-
Carbon Emissions
|
| 25 |
-
7 papers
|
| 26 |
-
Model card
|
| 27 |
-
Files and versions
|
| 28 |
-
Community
|
| 29 |
-
Settings
|
| 30 |
-
FAT5-small-UL2-fr
|
| 31 |
-
/ custom_heads_flash_t5.py
|
| 32 |
-
bourdoiscatie's picture
|
| 33 |
-
bourdoiscatie
|
| 34 |
-
Update custom_heads_flash_t5.py
|
| 35 |
-
3477ecb
|
| 36 |
-
verified
|
| 37 |
-
8 days ago
|
| 38 |
-
raw
|
| 39 |
-
history
|
| 40 |
-
blame
|
| 41 |
-
edit
|
| 42 |
-
delete
|
| 43 |
-
No virus
|
| 44 |
-
16.9 kB
|
| 45 |
import torch
|
| 46 |
import torch.nn as nn
|
| 47 |
from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
|
|
@@ -441,4 +397,4 @@ class FlashT5ForQuestionAnswering(FlashT5PreTrainedModel):
|
|
| 441 |
end_logits=end_logits,
|
| 442 |
hidden_states=outputs.hidden_states,
|
| 443 |
attentions=outputs.attentions,
|
| 444 |
-
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import torch
|
| 2 |
import torch.nn as nn
|
| 3 |
from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
|
|
|
|
| 397 |
end_logits=end_logits,
|
| 398 |
hidden_states=outputs.hidden_states,
|
| 399 |
attentions=outputs.attentions,
|
| 400 |
+
)
|