llamaindex
/

vdr-2b-v1

Image-Text-to-Text

sentence-transformers

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Delete self

#2

by antonioloison - opened 5 days ago

base: refs/heads/main

←

from: refs/pr/2

Discussion Files changed

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -131,7 +131,7 @@ def encode_queries(queries: list[str], dimension: int) -> torch.Tensor:
         **inputs, cache_position=cache_position, use_cache=False)
     with torch.no_grad():
-        output = self.model(
             **inputs,
             return_dict=True,
             output_hidden_states=True
@@ -195,7 +195,7 @@ def encode_documents(documents: list[Image.Image], dimension: int):
         **inputs, cache_position=cache_position, use_cache=False)
     with torch.no_grad():
-        output = self.model(
             **inputs,
             return_dict=True,
             output_hidden_states=True

         **inputs, cache_position=cache_position, use_cache=False)
     with torch.no_grad():
+        output = model(
             **inputs,
             return_dict=True,
             output_hidden_states=True
         **inputs, cache_position=cache_position, use_cache=False)
     with torch.no_grad():
+        output = model(
             **inputs,
             return_dict=True,
             output_hidden_states=True