Add `save_in_root` to the custom Module to allow saving & loading locally

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',