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
|