ydshieh HF staff ragavsachdeva commited on
Commit
bfb1631
1 Parent(s): 48e3ede

Add missing import (#18)

Browse files

- Add missing import (6e0ccc6dfd0f1c4b90b060a5028228d8b6e6c917)


Co-authored-by: Ragav Sachdeva <[email protected]>

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 (