Add missing import
#18
by
ragavsachdeva
- opened
- 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 (
|