Add `save_in_root` to the custom Module to allow saving & loading locally (#17)

- Add `save_in_root` to the custom Module to allow saving & loading locally (d901c3c1918acb56f0dbbe68ec35b34960057045)


Co-authored-by: Tom Aarsen <[email protected]>

custom_st.py

@@ -12,6 +12,9 @@ from transformers import AutoConfig, AutoImageProcessor, AutoModel, AutoTokenize
 
 
 class Transformer(nn.Module):
+
+    save_in_root: bool = True
+    
     def __init__(
         self,
         model_name_or_path: str = 'jinaai/jina-clip-v2',