refactor-remove-is-single (#10)
Browse files- refactor: remove checking for single inputs (3dd1f9db83b070294f7fd184ba3bf2fb5dad46b5)
modeling_jina_embeddings_v4.py
CHANGED
@@ -398,7 +398,6 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
|
|
398 |
prefix=encode_kwargs.pop("prefix"),
|
399 |
)
|
400 |
|
401 |
-
is_single = len(texts) == 1
|
402 |
embeddings = self._process_batches(
|
403 |
data=texts,
|
404 |
processor_fn=processor_fn,
|
@@ -408,7 +407,7 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
|
|
408 |
**encode_kwargs,
|
409 |
)
|
410 |
|
411 |
-
return embeddings
|
412 |
|
413 |
def encode_images(
|
414 |
self,
|
@@ -439,7 +438,6 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
|
|
439 |
|
440 |
encode_kwargs = self._validate_encoding_params(vector_type, truncate_dim)
|
441 |
|
442 |
-
is_single = len(images) == 1
|
443 |
embeddings = self._process_batches(
|
444 |
data=images,
|
445 |
processor_fn=self.processor.process_images,
|
@@ -452,7 +450,7 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
|
|
452 |
if max_pixels:
|
453 |
self.processor.image_processor.max_pixels = default_max_pixels
|
454 |
|
455 |
-
return embeddings
|
456 |
|
457 |
@classmethod
|
458 |
def from_pretrained(
|
|
|
398 |
prefix=encode_kwargs.pop("prefix"),
|
399 |
)
|
400 |
|
|
|
401 |
embeddings = self._process_batches(
|
402 |
data=texts,
|
403 |
processor_fn=processor_fn,
|
|
|
407 |
**encode_kwargs,
|
408 |
)
|
409 |
|
410 |
+
return embeddings
|
411 |
|
412 |
def encode_images(
|
413 |
self,
|
|
|
438 |
|
439 |
encode_kwargs = self._validate_encoding_params(vector_type, truncate_dim)
|
440 |
|
|
|
441 |
embeddings = self._process_batches(
|
442 |
data=images,
|
443 |
processor_fn=self.processor.process_images,
|
|
|
450 |
if max_pixels:
|
451 |
self.processor.image_processor.max_pixels = default_max_pixels
|
452 |
|
453 |
+
return embeddings
|
454 |
|
455 |
@classmethod
|
456 |
def from_pretrained(
|