Upload test_load_large.py

test_load_large.py

@@ -0,0 +1,18 @@
+from safetensors.torch import load_file
+from models.InternVideo_next import internvideo_next_large_patch14_224
+
+# test loading
+test_model = internvideo_next_large_patch14_224()
+model_ckpt = load_file('internvideo_next_large_ps14_res224.safetensors')
+
+msg = test_model.load_state_dict(model_ckpt, strict=False)
+print(msg)
+
+# test input
+import torch
+
+test_model = test_model.cuda().half()
+input_data = torch.randn(16, 3, 16, 224, 224).cuda().half() # B, C, T, H, W
+output_embedding = test_model(input_data)
+
+print(output_embedding.shape) # [16, 4096, 1024]