fix bug image_processing
Browse files
image_processing_minicpmv.py
CHANGED
@@ -52,6 +52,7 @@ class MiniCPMVBatchFeature(BatchFeature):
|
|
52 |
if not is_tensor(value):
|
53 |
tensor = as_tensor(value)
|
54 |
return tensor
|
|
|
55 |
except: # noqa E722
|
56 |
if key == "overflowing_values":
|
57 |
raise ValueError("Unable to create tensor returning overflowing values of different lengths. ")
|
@@ -71,6 +72,8 @@ class MiniCPMVBatchFeature(BatchFeature):
|
|
71 |
|
72 |
def cast_tensor(v):
|
73 |
# check if v is a floating point
|
|
|
|
|
74 |
if torch.is_floating_point(v):
|
75 |
# cast and send to device
|
76 |
return v.to(*args, **kwargs)
|
|
|
52 |
if not is_tensor(value):
|
53 |
tensor = as_tensor(value)
|
54 |
return tensor
|
55 |
+
return value
|
56 |
except: # noqa E722
|
57 |
if key == "overflowing_values":
|
58 |
raise ValueError("Unable to create tensor returning overflowing values of different lengths. ")
|
|
|
72 |
|
73 |
def cast_tensor(v):
|
74 |
# check if v is a floating point
|
75 |
+
if v is None:
|
76 |
+
return None
|
77 |
if torch.is_floating_point(v):
|
78 |
# cast and send to device
|
79 |
return v.to(*args, **kwargs)
|