Add missing import (#18)

- Add missing import (6e0ccc6dfd0f1c4b90b060a5028228d8b6e6c917)


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

modeling_kosmos2.py

@@ -22,6 +22,7 @@ from typing import List, Optional, Tuple, Union
 import torch
 import torch.utils.checkpoint
 from torch import nn
+from torch.nn import CrossEntropyLoss
 
 from transformers.activations import ACT2FN
 from transformers.modeling_outputs import (