Commit
•
6e0ccc6
1
Parent(s):
48e3ede
Add missing import
Browse files- modeling_kosmos2.py +1 -0
modeling_kosmos2.py
CHANGED
@@ -22,6 +22,7 @@ from typing import List, Optional, Tuple, Union
|
|
22 |
import torch
|
23 |
import torch.utils.checkpoint
|
24 |
from torch import nn
|
|
|
25 |
|
26 |
from transformers.activations import ACT2FN
|
27 |
from transformers.modeling_outputs import (
|
|
|
22 |
import torch
|
23 |
import torch.utils.checkpoint
|
24 |
from torch import nn
|
25 |
+
from torch.nn import CrossEntropyLoss
|
26 |
|
27 |
from transformers.activations import ACT2FN
|
28 |
from transformers.modeling_outputs import (
|