Files changed (1) hide show
  1. 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 (