Update custom_st.py to be compatible with Python 3.8 (#55)

- Update custom_st.py to be compatible with Python 3.8 (e20f32ee20be7027372c7a14a95892829fa42ae9)


Co-authored-by: pewhpewh <[email protected]>

custom_st.py

@@ -190,7 +190,7 @@ class Transformer(nn.Module):
         )
         return output
 
-    def get_config_dict(self) -> dict[str, Any]:
+    def get_config_dict(self) -> Dict[str, Any]:
          return {key: self.__dict__[key] for key in self.config_keys}
 
     def save(self, output_path: str, safe_serialization: bool = True) -> None: