finalf0 commited on
Commit
9d4cbd2
1 Parent(s): b68e5e1

remove timm

Browse files
Files changed (1) hide show
  1. modeling_minicpmv.py +2 -1
modeling_minicpmv.py CHANGED
@@ -5,7 +5,6 @@ import torch
5
  import torchvision
6
  from copy import deepcopy
7
  from PIL import Image
8
- from timm.data import IMAGENET_INCEPTION_MEAN, IMAGENET_INCEPTION_STD
9
  from torchvision import transforms
10
  from transformers import LlamaTokenizer, LlamaPreTrainedModel, LlamaForCausalLM, AutoModel, PreTrainedTokenizerFast
11
  from transformers.models.idefics2.modeling_idefics2 import Idefics2VisionTransformer
@@ -13,6 +12,8 @@ from transformers.models.idefics2.modeling_idefics2 import Idefics2VisionTransfo
13
  from .configuration_minicpm import MiniCPMVConfig
14
  from .resampler import Resampler
15
 
 
 
16
 
17
  class MiniCPMVPreTrainedModel(LlamaPreTrainedModel):
18
  config_class = MiniCPMVConfig
 
5
  import torchvision
6
  from copy import deepcopy
7
  from PIL import Image
 
8
  from torchvision import transforms
9
  from transformers import LlamaTokenizer, LlamaPreTrainedModel, LlamaForCausalLM, AutoModel, PreTrainedTokenizerFast
10
  from transformers.models.idefics2.modeling_idefics2 import Idefics2VisionTransformer
 
12
  from .configuration_minicpm import MiniCPMVConfig
13
  from .resampler import Resampler
14
 
15
+ IMAGENET_INCEPTION_MEAN = (0.5, 0.5, 0.5) # timm.data.IMAGENET_INCEPTION_MEAN
16
+ IMAGENET_INCEPTION_STD = (0.5, 0.5, 0.5) # timm.data.IMAGENET_INCEPTION_STD
17
 
18
  class MiniCPMVPreTrainedModel(LlamaPreTrainedModel):
19
  config_class = MiniCPMVConfig