Add reference to config class

model_flowformer.py

@@ -71,6 +71,8 @@ class Flowformer(PreTrainedModel):
             Initializing with a config file does not load the weights associated with the model, only the
             configuration. Check out the [`~PreTrainedModel.from_pretrained`] method to load the model weights.
     """
+    config_class = FlowformerConfig
+    
     def __init__(self, config: FlowformerConfig):
         super().__init__(config)