DmitryInd commited on
Commit
c385ec4
·
verified ·
1 Parent(s): e34cd13

Update modeling_cogvlm.py

Browse files

Update of modeling_cogvlm.py for Transformers newer version (copy of the similar commit in THUDM/cogvlm2-llama3-chat-19B)

Files changed (1) hide show
  1. modeling_cogvlm.py +1 -0
modeling_cogvlm.py CHANGED
@@ -5,6 +5,7 @@ from typing import TYPE_CHECKING, Optional, Tuple, List, Union, Literal, Dict, A
5
 
6
  import math
7
  import torch
 
8
  from torch import nn
9
  from torch.nn import CrossEntropyLoss
10
  from torchvision import transforms
 
5
 
6
  import math
7
  import torch
8
+ import transformers
9
  from torch import nn
10
  from torch.nn import CrossEntropyLoss
11
  from torchvision import transforms