remove timm
Browse files- 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
|