llamaindex
/

vdr-2b-v1

sentence-transformers

text-generation-inference

Model card Files Files and versions

cheesyFishes commited on Jan 10

Commit

f6ad57e

·

verified ·

1 Parent(s): 0965666

Update custom_st.py

Files changed (1) hide show

custom_st.py +1 -1

custom_st.py CHANGED Viewed

@@ -17,7 +17,7 @@ class Transformer(nn.Module):
     def __init__(
         self,
-        model_name_or_path: str = 'llamaindex/vdr-2b-multi-v1',
         processor_name_or_path: Optional[str] = None,
         max_pixels: int = 768 * 28 * 28,
         min_pixels: int = 1 * 28 * 28,

     def __init__(
         self,
+        model_name_or_path: str = 'llamaindex/vdr-2b-v1',
         processor_name_or_path: Optional[str] = None,
         max_pixels: int = 768 * 28 * 28,
         min_pixels: int = 1 * 28 * 28,