dahwinsingularity commited on
Commit
9449531
1 Parent(s): edf99e6

Update modeling_minicpmv.py

Browse files
Files changed (1) hide show
  1. modeling_minicpmv.py +1 -0
modeling_minicpmv.py CHANGED
@@ -9,6 +9,7 @@ from PIL import Image
9
  from torchvision import transforms
10
  from transformers import LlamaTokenizer, LlamaPreTrainedModel, LlamaForCausalLM, AutoModel, PreTrainedTokenizerFast
11
  from transformers.models.idefics2.modeling_idefics2 import Idefics2VisionTransformer
 
12
 
13
  from .configuration_minicpm import MiniCPMVConfig
14
  from .resampler import Resampler
 
9
  from torchvision import transforms
10
  from transformers import LlamaTokenizer, LlamaPreTrainedModel, LlamaForCausalLM, AutoModel, PreTrainedTokenizerFast
11
  from transformers.models.idefics2.modeling_idefics2 import Idefics2VisionTransformer
12
+ import torch.nn as nn
13
 
14
  from .configuration_minicpm import MiniCPMVConfig
15
  from .resampler import Resampler