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
|