decord import bugfix in VLMs

zanedurante · Nov 14, 2022 · 0b50a19 · 0b50a19
1 parent 44990e0
commit 0b50a19
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/MILES/wrapper.py b/MILES/wrapper.py
@@ -1,10 +1,10 @@
 import os, sys
 from typing import Optional, List
 import numpy as np
-import decord
 from transformers import AutoTokenizer
 
 import torch
+import decord
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 from SimilarityVLM import SimilarityVLM

diff --git a/UNIVL/wrapper.py b/UNIVL/wrapper.py
@@ -3,11 +3,11 @@
 from types import SimpleNamespace
 import pandas as pd
 import numpy as np
-import decord
 from typing import Optional
 from torchvision.transforms import Compose, CenterCrop, Resize, Lambda
 
 import torch
+import decord
 import torch.nn.functional as F
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 

diff --git a/VTTWINS/wrapper.py b/VTTWINS/wrapper.py
@@ -1,9 +1,9 @@
 import importlib
 import numpy as np
 import os, sys
-import decord
 
 import torch
+import decord
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 from SimilarityVLM import *