Update README.md
Browse files
README.md
CHANGED
@@ -119,6 +119,8 @@ model_path = 'OpenGVLab/InternVideo2_5_Chat_8B'
|
|
119 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
120 |
model = AutoModel.from_pretrained(model_path, trust_remote_code=True).half().cuda()
|
121 |
|
|
|
|
|
122 |
|
123 |
def build_transform(input_size):
|
124 |
MEAN, STD = IMAGENET_MEAN, IMAGENET_STD
|
|
|
119 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
120 |
model = AutoModel.from_pretrained(model_path, trust_remote_code=True).half().cuda()
|
121 |
|
122 |
+
IMAGENET_MEAN = (0.485, 0.456, 0.406)
|
123 |
+
IMAGENET_STD = (0.229, 0.224, 0.225)
|
124 |
|
125 |
def build_transform(input_size):
|
126 |
MEAN, STD = IMAGENET_MEAN, IMAGENET_STD
|